cupy.shape

cupy.shape(a)[source]

Returns the shape of an array

Parameters

a (array_like) – Input array

Returns

The elements of the shape tuple give the lengths of the corresponding array dimensions.

Return type

tuple of ints