RavelMultiIndex

Converts a tuple of index arrays into an array of flat indices, applying boundary modes to the multi-index.

Abstract Signature:

RavelMultiIndex(multi_index: Tuple[Array], dims: Tuple[int], mode: str = raise, order: str = C)

PyTorchJAX (Core)NumPyTensorFlowFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Plugin (ravel_multi_index_plugin)

JAX (Core)

API: jax.numpy.ravel_multi_index

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.ravel_multi_index

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.ravel_multi_index

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: jax.numpy.ravel_multi_index

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.ravel_multi_index

Strategy: Direct Mapping

Official Docs ↗