jax.experimental.sparse.bcoo_sort_indices

jax.experimental.sparse.bcoo_sort_indices(mat)

排序 BCOO 数组的索引。

参数:

mat (BCOO) – BCOO 数组

返回:

索引已排序的 BCOO 数组。

返回类型:

mat_out