jax.experimental.sparse.bcoo_slice

jax.experimental.sparse.bcoo_slice(mat, *, start_indices, limit_indices, strides=None)

Sparse implementation of {func}`jax.lax.slice`.

参数:

• mat (BCOO) – 要重塑的 BCOO 数组。

• start_indices (Sequence[int]) – 长度为 mat.ndim 的整数序列，指定每个切片的起始索引。

• limit_indices (Sequence[int]) – 长度为 mat.ndim 的整数序列，指定每个切片的结束索引

• strides (Sequence[int] | None | None) – （未实现）长度为 mat.ndim 的整数序列，指定每个切片的步长

返回:

包含该切片的 BCOO 数组。

返回类型:

out