tf.math.cumsum

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

tf.math.cumsum(
    x, axis=0, 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]