jax.lax.squeeze

目录

jax.lax.squeeze#

jax.lax.squeeze(array, dimensions)[源代码]#

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

参数：

array (ArrayLike)
dimensions (Sequence[int])

返回类型：