jax.Array.sum#

abstractArray.sum(axis=None,dtype=None,out=None,keepdims=False,initial=None,where=None,promote_integers=True)[source]#

Sum of the elements of the array over a given axis.

Refer tojax.numpy.sum() for full documentation.

Parameters:

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike |None)

• out (None)

• keepdims (bool)

• initial (ArrayLike |None)

• where (ArrayLike |None)

• promote_integers (bool)

Return type:

Array