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]