maxframe.tensor.ndim

maxframe.tensor.ndim(a)

返回张量的维度数。

参数:

a (array_like) -- 输入张量。如果它还不是张量，则尝试进行转换。

返回:

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

返回类型:

int

参见

ndarray.ndim

等效方法

shape

张量的维度

Tensor.shape

张量的维度

示例

>>> import maxframe.tensor as mt
>>> mt.ndim([[1,2,3],[4,5,6]])
2
>>> mt.ndim(mt.array([[1,2,3],[4,5,6]]))
2
>>> mt.ndim(1)
0