jax.Array.squeeze

抽象 Array.squeeze(axis=None)

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

请参阅 jax.numpy.squeeze() 以获取完整文档。

参数:

• self (数组)

• axis (reductions.Axis)

返回类型:

数组