jax.Array.squeeze# abstract Array.squeeze(axis=None)[源代码]# 从数组中移除一个或多个长度为1的轴。 请参阅 jax.numpy.squeeze() 获取完整文档。 参数: self (Array) axis (reductions.Axis) 返回类型: Array
