jax.experimental.sparse.bcoo_squeeze

jax.experimental.sparse.bcoo_squeeze(arr, *, dimensions)

{func}`jax.lax.squeeze` 的稀疏實作。

從陣列中擠壓任意數量的尺寸為 1 的維度。

參數:

• arr (BCOO) – 要重新塑形的 BCOO 陣列。

• dimensions (Sequence[int]) – 指定要擠壓的維度的整數序列。

回傳值:

重新塑形的陣列。

回傳型別:

out