jax.experimental.sparse.bcoo_sort_indices

jax.experimental.sparse.bcoo_sort_indices#

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

Sort indices of a BCOO array.

Parameters:

mat (BCOO) – BCOO array

Returns:

BCOO array with sorted indices.

Return type:

mat_out