tf.math.accumulate_n  |  TensorFlow v2.0.0 (original) (raw)

tf.math.accumulate_n

Stay organized with collections Save and categorize content based on your preferences.

Returns the element-wise sum of a list of tensors.

View aliases

Compat aliases for migration

SeeMigration guide for more details.

tf.compat.v1.accumulate_n, tf.compat.v1.math.accumulate_n

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

Optionally, pass shape and tensor_dtype for shape and type checking, otherwise, these are inferred.

accumulate_n performs the same operation as tf.math.add_n, but does not wait for all of its inputs to be ready before beginning to sum. This approach can save memory if inputs are ready at different times, since minimum temporary storage is proportional to the output size rather than the inputs' size.

accumulate_n is differentiable (but wasn't previous to TensorFlow 1.7).

For example:

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