jax.experimental.sparse.bcoo_sort_indices

jax.experimental.sparse.bcoo_sort_indices(mat)

排序 BCOO 陣列的索引。

參數:

mat (BCOO) – BCOO 陣列

傳回:

具有排序索引的 BCOO 陣列。

傳回類型:

mat_out