jax.experimental.sparse.bcoo_sum_duplicates#

jax.experimental.sparse.bcoo_sum_duplicates(mat, nse=None)[來源]#

加總 BCOO 陣列內的重複索引，並傳回具有排序索引的陣列。

參數:

  • mat (BCOO) – BCOO 陣列

  • nse (int | None | None) – 整數（選用）。輸出矩陣中指定元素的數量。為了使 bcoo_sum_duplicates 與 JIT 和其他 JAX 轉換相容，必須指定此值。如果未指定，將根據資料和索引陣列的內容計算最佳 nse。如果指定的 nse 大於必要值，資料和索引陣列將以標準填充值填充。如果小於必要值，資料元素將從輸出矩陣中刪除。

傳回:

具有排序索引且沒有重複索引的 BCOO 陣列。

傳回型別:

mat_out