jax.experimental.sparse.bcsr_extract

jax.experimental.sparse.bcsr_extract#

jax.experimental.sparse.bcsr_extract(indices, indptr, mat)[源代码][源代码]#

从给定的BCSR（索引，indptr）中提取密集矩阵的值。

参数:

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

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

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

返回:

一个 ndarray；参见 BCSR 数据。

返回类型:

Array

目录