jax.experimental.sparse.bcsr_extract#

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

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

参数:

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

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

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

返回:

一个 ndarray；请参阅 BCSR data。

返回类型:

Array