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