jax.Array.sum

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

沿給定軸計算陣列元素的總和。

完整文件請參閱 jax.numpy.sum()。

參數:

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike | None)

• out (None)

• keepdims (bool)

• initial (ArrayLike | None)

• where (ArrayLike | None)

• promote_integers (bool)

回傳型別:

Array