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