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, но не ожидает, что все входные данные будут готовы перед началом суммирования. Этот подход может сэкономить память, если входные данные готовы в разное время, поскольку минимальное временное хранилище пропорционально размеру вывода, а не размеру входных данных.

accumulate_n дифференцируема (но не была до TensorFlow 1.7).

Например:

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]]