jax.Array.cumsum#

abstractArray.cumsum(axis=None,dtype=None,out=None)[source]#

Return the cumulative sum of the array.

Refer tojax.numpy.cumsum() for the full documentation.

Parameters:

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike |None)

• out (None)

Return type:

Array