Indices

Return an array representing the indices of a grid.

Abstract Signature:

Indices(dimensions: Tuple[int], dtype: DType = int, sparse: bool = False)

PyTorchJAX (Core)NumPyTensorFlowApple MLXFlax NNX

PyTorch

API: —

Strategy: Custom / Partial

JAX (Core)

API: jax.numpy.indices

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.indices

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.meshgrid

Strategy: Plugin (indices_shim)

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: jax.numpy.indices

Strategy: Direct Mapping

Official Docs ↗