tf.keras.backend.cumsum

Cumulative sum of the values in a tensor, alongside the specified axis.

x A tensor or variable.
axis An integer, the axis to compute the sum.

A tensor of the cumulative sum of values of x along axis.