jax.experimental.sparse.bcoo_gather#
- jax.experimental.sparse.bcoo_gather(operand, start_indices, dimension_numbers, slice_sizes, *, unique_indices=False, indices_are_sorted=False, mode=None, fill_value=None)[源代码][源代码]#
lax.gather 的 BCOO 版本。
- 参数:
operand (BCOO)
start_indices (Array)
dimension_numbers (GatherDimensionNumbers)
slice_sizes (Shape)
unique_indices (bool)
indices_are_sorted (bool)
mode (str | GatherScatterMode | None)
- 返回类型: