jax.Array.squeeze

abstract Array.squeeze(axis=None)

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

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

参数:

• self (Array)

• axis (reductions.Axis)

返回类型:

Array