jax.numpy.shape

jax.numpy.shape(a)

返回数组的形状。

参数:

a (array_like) – 输入数组。

返回:

shape – shape 元组的元素给出了相应数组维度的长度。

返回类型:

tuple of ints

参见

len

对于 N>=1 的 N-D 数组，len(a) 等同于 np.shape(a)[0]。

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,)