jax.numpy.ndim ============== .. currentmodule:: jax.numpy .. autofunction:: ndim