jax.experimental.sparse.bcoo_slice

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

jax.lax.slice 的稀疏实现。

参数:

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

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

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

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

返回:

包含切片的 BCOO 数组。

返回类型:

返回值