jax.experimental.sparse.bcsr_todense

jax.experimental.sparse.bcsr_todense#

jax.experimental.sparse.bcsr_todense(mat)[source]#

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

参数:

**mat** (BCSR) – BCSR 矩阵。

返回:

mat 的密集版本。

返回类型:

Array