jax.experimental.sparse.bcsr_fromdense#

jax.experimental.sparse.bcsr_fromdense(mat, *, nse=None, n_batch=0, n_dense=0, index_dtype=<class 'jax.numpy.int32'>)[source]#

Create BCSR-format sparse matrix from a dense matrix.

Parameters:
Returns:

BCSR representation of the matrix.

Return type:

mat_bcsr