main/transformer_utils/mmpose/models/utils/misc.py

# Copyright (c) OpenMMLab. All rights reserved.
import torch
from packaging import version

_torch_version_meshgrid_indexing = version.parse(
    torch.__version__) >= version.parse('1.10.0a0')


def torch_meshgrid_ij(*tensors):
    if _torch_version_meshgrid_indexing:
        return torch.meshgrid(*tensors, indexing='ij')
    else:
        return torch.meshgrid(*tensors)  # Uses indexing='ij' by default