mlx.core.squeeze

squeeze(a: array, /, axis: None | int | Sequence[int] = None, *, stream: None | Stream | Device = None) → array

Remove length one axes from an array.

參數:

• a (array) -- Input array.

• axis (int or tuple(int), optional) -- Axes to remove. Defaults to None in which case all size one axes are removed.

回傳:

The output array with size one axes removed.

回傳型別:

array