jax.experimental.sparse.bcsr_extract#

jax.experimental.sparse.bcsr_extract(indices, indptr, mat)[source]#

从给定的 BCSR (indices, indptr) 的密集矩阵中提取值。

参数:

  • indices (ArrayLike) – 一个 ndarray; 参见 BCSR 索引。

  • indptr (ArrayLike) – 一个 ndarray; 参见 BCSR indptr。

  • mat (ArrayLike) – 一个密集矩阵。

返回:

一个 ndarray; 参见 BCSR 数据。

返回类型:

数组