jax.lax.squeeze

内容

jax.lax.squeeze#

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

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

参数：:

array (ArrayLike)
dimensions (Sequence[int])

返回类型：: