jax.lax.squeeze#

jax.lax.squeeze(array, dimensions)[source]#

从数组中挤压任意数量的大小为 1 的维度。

参数:

• array (ArrayLike)

• dimensions (Sequence[int])

返回类型:

Array