jax.numpy.shape

jax.numpy.shape(a)

返回数组的形状。

参数:

a (array_like) – 输入数组。

返回值:

shape – 形状元组的元素给出对应数组维度的长度。

返回类型:

tuple of ints

参见

len

len(a) 等效于 np.shape(a)[0]，对于具有 N>=1 的 N 维数组。

ndarray.shape

等效的数组方法。

示例

>>> import numpy as np
>>> np.shape(np.eye(3))
(3, 3)
>>> np.shape([[1, 3]])
(1, 2)
>>> np.shape([0])
(1,)
>>> np.shape(0)
()

>>> a = np.array([(1, 2), (3, 4), (5, 6)],
...              dtype=[('x', 'i4'), ('y', 'i4')])
>>> np.shape(a)
(3,)
>>> a.shape
(3,)