tf.math.add_n

Складывает все входные тензоры поэлементно.

tf.math.add_n(
    inputs, name=None
)

Преобразует объекты IndexedSlices в плотные тензоры перед сложением.

tf.math.add_n выполняет ту же операцию, что и tf.math.accumulate_n, но ожидает, пока все её входы будут готовы, прежде чем начать суммирование. Это буферизация может привести к большему потреблению памяти, когда входы готовы в разное время, так как минимальный временный объем памяти пропорционален размеру входных данных, а не размеру выходных данных.

Эта операция не распространяется на свои входные данные. Если вам нужна трансляция, используйте tf.math.add (или оператор +).

Например:

a = tf.constant([[3, 5], [4, 8]])
b = tf.constant([[1, 6], [2, 9]])
tf.math.add_n([a, b, a])  # [[7, 16], [10, 25]]