jax.Array.squeeze

abstract Array.squeeze(axis=None)

从数组中删除一个或多个长度为 1 的轴。

有关完整文档，请参阅 jax.numpy.squeeze()。

参数：

• self (Array)

• axis (reductions.Axis)

返回类型：

Array