jax.numpy.ndim

jax.numpy.ndim(a)

返回数组的维度数。

参数:

a (array_like) – 输入数组。如果它不是 ndarray，则尝试进行转换。

返回值:

number_of_dimensions – a 中的维度数。标量是零维。

返回类型:

int

另请参见

ndarray.ndim

等效方法

shape

数组的维度

ndarray.shape

数组的维度

示例

>>> import numpy as np
>>> np.ndim([[1,2,3],[4,5,6]])
2
>>> np.ndim(np.array([[1,2,3],[4,5,6]]))
2
>>> np.ndim(1)
0