jax.experimental.sparse.bcoo_dynamic_slice

jax.experimental.sparse.bcoo_dynamic_slice(mat, start_indices, slice_sizes)

稀疏 {func}`jax.lax.dynamic_slice` 的實作。

參數:

• mat (BCOO) – 要切片的 BCOO 陣列。

• start_indices (Sequence[Any]) – 每個維度一個純量索引的列表。這些值可以是動態的。

• slice_sizes (Sequence[int]) – 切片的大小。必須是非負整數序列，長度等於 ndim(operand)。在 JIT 編譯函數內部，僅支援靜態值 (JIT 內的所有 JAX 陣列都必須具有靜態已知大小)。

返回:

包含切片的 BCOO 陣列。

返回類型:

out