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.
|
登录
TensorFlow 1 version
View source on GitHub