jax.experimental.sparse.csr_todense

jax.experimental.sparse.csr_todense(mat)

将 CSR 格式的稀疏矩阵转换为密集矩阵。

参数:

mat (CSR) – CSR 矩阵

返回:

mat 的密集版本

返回类型:

mat_dense