jax.experimental.sparse.bcoo_reduce_sum

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

在给定轴上对数组元素求和。

参数：

• mat (BCOO) – 一个 BCOO 格式的数组。

• shape – 目标数组的形状。

• axes (Sequence[int]) – 一个元组、列表或 ndarray，包含要对其执行求和的 mat 的轴。

返回值：

包含结果的 BCOO 格式数组。

返回类型：

BCOO