jax.experimental.sparse.bcoo_extract

jax.experimental.sparse.bcoo_extract(sparr, arr, *, assume_unique=None)

根据稀疏数组的索引从密集数组中提取值。

参数：

• sparr (BCOO) – BCOO 数组，其索引将用于输出。

• arr (ArrayLike) – 形状等于 self.shape 的 ArrayLike

• assume_unique (bool | None | None) – 布尔值，默认为 sparr.unique_indices。如果为 True，则提取每个索引的值，即使索引包含重复项。如果为 False，则重复索引的值将被求和并返回到第一个索引的位置。

返回：

一个具有与 self 相同稀疏模式的 BCOO 数组。

返回类型：

已提取