jax.experimental.sparse.bcoo_sort_indices

jax.experimental.sparse.bcoo_sort_indices#

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

对 BCOO 数组的索引进行排序。

参数:

mat (BCOO) – BCOO 数组

返回值:

具有已排序索引的 BCOO 数组。

返回类型:

mat_out