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