jax.experimental.sparse.bcoo_extract

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

根據稀疏陣列的索引，從密集陣列中提取值。

參數:

• sparr (BCOO) – BCOO 陣列，其索引將用於輸出。

• arr (ArrayLike) – ArrayLike，形狀等於 self.shape

• assume_unique (bool | None | None) – bool，預設為 sparr.unique_indices。如果為 True，則為每個索引提取值，即使索引包含重複項。如果為 False，重複的索引將使其值加總，並在第一個索引的位置傳回。

傳回:

一個 BCOO 陣列，其稀疏模式與 self 相同。

傳回型別:

extracted