jax.experimental.sparse.bcoo_gather

内容

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)[source]#

BCOO 版本的 lax.gather。

参数:
返回类型:

BCOO