jax.experimental.sparse.bcoo_排序_索引

jax.experimental.sparse.bcoo_排序_索引#

jax.experimental.sparse.bcoo_sort_indices(mat)[源代码][源代码]#

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

参数:

mat (BCOO) – BCOO 数组

返回:

带有排序索引的 BCOO 数组。

返回类型:

mat_out