jax.experimental.sparse.bcsr_extract

jax.experimental.sparse.bcsr_extract#

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

从给定 BCSR（indices，indptr）的稠密矩阵中提取值。

参数：

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

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

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

返回值：

一个 ndarray；参见 BCSR data。

返回类型：

Array

内容