Shortcuts

mmengine.utils.dl_utils.torch_meshgrid

mmengine.utils.dl_utils.torch_meshgrid(*tensors)[source]

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.

Parameters

tensors (List[Tensor]) – List of scalars or 1 dimensional tensors.

Returns

Sequence of meshgrid tensors.

Return type

Sequence[Tensor]

Read the Docs v: v0.7.0
Versions
latest
stable
v0.7.0
v0.6.0
v0.5.0
v0.4.0
v0.3.0
v0.2.0
Downloads
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.