cupy.shape

cupy.shape(a)

返回数组的形状

参数:

a (array_like) – 输入数组

返回:

形状元组的元素表示相应数组维度的长度。

返回类型:

整数的 tuple