jax.experimental.sparse.bcoo_slice

目录

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) – (未实现) 长度为 mat.ndim 的整数序列,用于指定每个切片的步幅

返回:

包含切片信息的 BCOO 数组。

返回类型:

out