mmengine.utils.dl_utils.torch_meshgrid¶
- mmengine.utils.dl_utils.torch_meshgrid(*tensors)[源代码]¶
A wrapper of torch.meshgrid to compat different PyTorch versions.
Since PyTorch 1.10.0a0, torch.meshgrid supports the arguments
indexing
. So we implement a wrapper here to avoid warning when using high-version PyTorch and avoid compatibility issues when using previous versions of PyTorch.- 参数:
tensors (List[Tensor]) – List of scalars or 1 dimensional tensors.
- 返回:
Sequence of meshgrid tensors.
- 返回类型:
Sequence[Tensor]