Registration is open for TensorFlow Dev Summit 2020 Learn more

tf.math.accumulate_n

TensorFlow 1 version View source on GitHub

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

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.

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

Args:

  • inputs: A list of Tensor objects, each with same shape and type.
  • shape: Expected shape of elements of inputs (optional). Also controls the output shape of this op, which may affect type inference in other ops. A value of None means "infer the input shape from the shapes in inputs".
  • tensor_dtype: Expected data type of inputs (optional). A value of None means "infer the input dtype from inputs[0]".
  • name: A name for the operation (optional).

Returns:

A Tensor of same shape and type as the elements of inputs.

Raises:

  • ValueError: If inputs don't all have same shape and dtype or the shape cannot be inferred.

Compat aliases