jax.lax.cumsum

Warning

This page was created from a pull request (#9655).

jax.lax.cumsum¶

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

Computes a cumulative sum along axis.

Parameters

operand (Any) –
axis (int) –
reverse (bool) –

Return type

previous

jax.lax.cumprod

next

jax.lax.digamma