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
