jax.experimental.sparse.csr_todense

jax.experimental.sparse.csr_todense(mat)

Convert a CSR-format sparse matrix to a dense matrix.

Parameters:

mat (CSR) – CSR matrix

Returns:

dense version of mat

Return type:

mat_dense