jax.lax.cumsum

Contents

jax.lax.cumsum#

jax.lax.cumsum(operand, axis=0, reverse=False)[source]#

Computes a cumulative sum along axis.

Parameters:

operand (Array)
axis (int)
reverse (bool)

Return type: