NdimΒΆ

Return the number of dimensions of an array.

Abstract Signature:

Ndim(a: Tensor)

PyTorch

API: ndim
Strategy: Macro '{a}.ndim'

JAX (Core)

API: jax.numpy.ndim
Strategy: Direct Mapping

NumPy

API: numpy.ndim
Strategy: Direct Mapping

Keras

API: ndim
Strategy: Macro 'len({a}.shape)'

TensorFlow

API: tf.rank
Strategy: Macro '{a}.ndim'

Apple MLX

API: ndim
Strategy: Macro '{a}.ndim'

Flax NNX

API: jax.numpy.ndim
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.ndim
Strategy: Direct Mapping