cupy.squeeze#
- cupy.squeeze(a, axis=None)[source]#
Removes size-one axes from the shape of an array.
- Parameters:
a (cupy.ndarray) – Array to be reshaped.
axis (int or tuple of ints) – Axes to be removed. This function removes all size-one axes by default. If one of the specified axes is not of size one, an exception is raised.
- Returns:
An array without (specified) size-one axes.
- Return type:
See also