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`

.

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

© 2020 The TensorFlow Authors. All rights reserved.

Licensed under the Creative Commons Attribution License 3.0.

Code samples licensed under the Apache 2.0 License.

https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/math/accumulate_n