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