jax.experimental.sparse.bcoo_sort_indices

jax.experimental.sparse.bcoo_sort_indices(mat)

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

参数:

mat (BCOO) – BCOO 数组

返回:

具有排序索引的 BCOO 数组。

返回类型:

mat_out