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