jax.lax.squeeze

jax.lax.squeeze(array, dimensions)

从数组中移除任意数量的大小为1的维度。

参数:

• array (ArrayLike)

• dimensions (Sequence[int])

返回类型:

Array