jax.experimental.sparse.bcoo_sum_duplicates# jax.experimental.sparse.bcoo_sum_duplicates(mat, nse=None)[源代码][源代码]# 对BCOO数组中的重复索引进行求和，返回一个索引已排序的数组。 参数: mat (BCOO) – BCOO 数组 nse (int | None) – 整数（可选）。输出矩阵中指定元素的数量。为了使 bcoo_sum_duplicates 与 JIT 和其他 JAX 转换兼容，必须指定此项。如果未指定，将根据数据和索引数组的内容计算最佳 nse。如果指定的 nse 大于必要值，数据和索引数组将用标准填充值填充。如果小于必要值，数据元素将从输出矩阵中删除。 返回: 具有排序索引且无重复索引的 BCOO 数组。 返回类型: mat_out
