cupy.squeeze

cupy.squeeze(a, axis=None)

从数组的形状中移除大小为一的轴。

参数:

• a (cupy.ndarray) – 要重塑的数组。

• axis (int 或 整数元组) – 要移除的轴。此函数默认移除所有大小为一的轴。如果指定的某个轴的大小不是一，则会引发异常。

返回:

一个移除了（指定）大小为一的轴的数组。

返回类型:

cupy.ndarray

另请参阅

numpy.squeeze()