tf.raw_ops.Cumsum

Вычисляет накопительную сумму тензора x вдоль axis.

tf.raw_ops.Cumsum(
    x, axis, exclusive=False, reverse=False, name=None
)

По умолчанию эта операция вычисляет включительную накопительную сумму, то есть первый элемент входных данных совпадает с первым элементом выходных данных:

tf.cumsum([a, b, c])  # => [a, a + b, a + b + c]

Если задать для аргумента exclusive значение True, вместо этого будет вычислена исключительная накопительная сумма:

tf.cumsum([a, b, c], exclusive=True)  # => [0, a, a + b]

Если задать для аргумента reverse значение True, накопительная сумма будет вычисляться в обратном направлении:

tf.cumsum([a, b, c], reverse=True)  # => [a + b + c, b + c, c]

Это эффективнее, чем использовать отдельные операции tf.reverse.

Аргументы reverse и exclusive также можно комбинировать:

tf.cumsum([a, b, c], exclusive=True, reverse=True)  # => [b + c, c, 0]