jax.experimental.sparse.bcoo_squeeze# jax.experimental.sparse.bcoo_squeeze(arr, *, dimensions)[源代码][源代码]# 稀疏实现 {func}`jax.lax.squeeze`。 从数组中挤压任意数量的尺寸为1的维度。 参数: arr (BCOO) – 要重塑的 BCOO 数组。 dimensions (Sequence[int]) – 指定要压缩的维度的整数序列。 返回: 重塑后的数组。 返回类型: out
