jax.experimental.sparse.bcsr_extract#

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

从给定的 BCSR (索引, 指针) 中提取稠密矩阵的值。

参数:

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

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

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

返回:

一个 ndarray；请参阅 BCSR 数据。

返回类型:

Array