jax.experimental.sparse.bcoo_slice

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

{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