jax.experimental.sparse.bcoo_reduce_sum

jax.experimental.sparse.bcoo_reduce_sum(mat, *, axes)

在給定軸上對陣列元素求和。

參數:

• mat (BCOO) – BCOO 格式陣列。

• shape – 目標陣列的形狀。

• axes (Sequence[int]) – 包含要執行求和的 mat 軸的元組、列表或 ndarray。

返回:

包含結果的 BCOO 格式陣列。

返回類型:

BCOO