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