jax.lax.squeeze

jax.lax.squeeze(array, dimensions)

從陣列中移除任意數目的尺寸為 1 的維度。

參數:

• array (ArrayLike)

• dimensions (Sequence[int])

回傳類型:

Array