jax.Array.squeeze

abstract Array.squeeze(axis=None)

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

完整文档请参考jax.numpy.squeeze()。

参数:

• self (Array)

• axis (reductions.Axis)

返回类型:

Array