jax.lax.squeeze# jax.lax.squeeze(array, dimensions)[源代码][源代码]# 从数组中挤压任意数量的尺寸为1的维度。 参数: array (ArrayLike) dimensions (Sequence[int]) 返回类型: Array
