jax.experimental.sparse.bcsr_extract

jax.experimental.sparse.bcsr_extract(indices, indptr, mat)

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

参数:

• indices (类数组) – 一个 ndarray；参见 BCSR 索引。

• indptr (类数组) – 一个 ndarray；参见 BCSR indptr。

• mat (类数组) – 一个稠密矩阵。

返回:

一个 ndarray；参见 BCSR 数据。

返回类型:

数组