tf.math.accumulate_n
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. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/math/accumulate_n