tf.keras.ops.cumsum

Returns the cumulative sum of elements along a given axis.

x Input tensor.
axis Axis along which the cumulative sum is computed. By default the input is flattened.
dtype dtype of returned tensor. Defaults to x.dtype.

Output tensor.