jax.experimental.sparse.bcsr_todense

jax.experimental.sparse.bcsr_todense(mat)

将批处理的稀疏矩阵转换为稠密矩阵。

参数:

mat (BCSR) – BCSR 矩阵。

返回:

mat 的稠密版本。

返回类型:

数组