jax.experimental.sparse.bcsr_todense

Contents

jax.experimental.sparse.bcsr_todense#

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

Convert batched sparse matrix to a dense matrix.

Parameters:

mat (BCSR) – BCSR matrix.

Return type:

Array

Returns:

The dense version of mat.