jax.numpy.shape¶

jax.numpy.shape(a)¶

Return the shape of an array.

Parameters: a (array_like) – Input array.
Returns: shape – The elements of the shape tuple give the lengths of the corresponding array dimensions.
Return type: tuple of ints

See also

alen

Examples

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

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