mlx.core.squeeze

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

移除数组中长度为1的轴。

参数:

• a (array) – 输入数组。

• axis (int 或 tuple(int), 可选) – 要移除的轴。默认为 None，此时移除所有长度为1的轴。

返回值:

移除了长度为1的轴的输出数组。

返回类型:

array