ndim

Contents

ndim#

class brainunit.math.ndim(a)#

Return the number of dimensions of an array or Quantity.

Parameters:

a (saiunit.Quantity | Array | ndarray | bool | number | bool | int | float | complex) – Input array.

Returns:

out – Number of dimensions.

Return type:

int

Examples

>>> import jax.numpy as jnp
>>> import saiunit.math as sumath
>>> sumath.ndim(jnp.zeros((2, 3)))
2
>>> import saiunit as u
>>> sumath.ndim(u.Quantity(jnp.zeros((2, 3, 4)), unit=u.meter))
3
Contents