jax.lax.cumsum

jax.lax.cumsum(operand, axis=0, reverse=False)

Computes a cumulative sum along axis.

Parameters:

• operand (Array)

• axis (int)

• reverse (bool)

Return type:

Array