UnravelIndex

Convert flat indices into multi-dimensional indices.

Abstract Signature:

UnravelIndex(indices: Tensor, shape)

PyTorch

API: torch.unravel_index
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.unravel_index
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.unravel_index
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.unravel_index
Strategy: Direct Mapping
Official Docs ↗