jax.experimental.sparse.bcoo_todense

Contents

jax.experimental.sparse.bcoo_todense#

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

Convert batched sparse matrix to a dense matrix.

Parameters:

mat (BCOO) – BCOO matrix.

Returns:

dense version of mat.

Return type:

mat_dense