jax.Array.sum

抽象 Array.sum(axis=None, dtype=None, out=None, keepdims=False, initial=None, where=None, promote_integers=True)

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

有关完整文档，请参阅 jax.numpy.sum()。

参数:

• self (数组)

• axis (reductions.Axis)

• dtype (DTypeLike | None)

• out (None)

• keepdims (布尔值)

• initial (ArrayLike | None)

• where (ArrayLike | None)

• promote_integers (布尔值)

返回类型:

数组