tf.math.accumulate_n

Возвращает поэлементную сумму списка тензоров.

tf.math.accumulate_n(
    inputs, shape=None, tensor_dtype=None, name=None
)

Можно дополнительно передать shape и tensor_dtype для проверки формы и типа, в противном случае они определяются автоматически.

accumulate_n выполняет ту же операцию, что и tf.math.add_n.

Например:

a = tf.constant([[1, 2], [3, 4]])
b = tf.constant([[5, 0], [0, 6]])
tf.math.accumulate_n([a, b, a])  # [[7, 4], [6, 14]]

# Explicitly pass shape and type
tf.math.accumulate_n([a, b, a], shape=[2, 2], tensor_dtype=tf.int32)
                                                               # [[7,  4],
                                                               #  [6, 14]]