Returns the element-wise sum of a list of tensors.
tf.raw_ops.AccumulateNV2( inputs, shape, name=None )
tf.accumulate_n_v2
performs the same operation as tf.add_n
, but does not wait for all of its inputs to be ready before beginning to sum. This 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.
Unlike the original accumulate_n
, accumulate_n_v2
is differentiable.
Returns a Tensor
of same shape and type as the elements of inputs
.
Args | |
---|---|
inputs | A list of at least 1 Tensor objects with the same type in: float32 , float64 , int32 , uint8 , int16 , int8 , complex64 , int64 , qint8 , quint8 , qint32 , bfloat16 , uint16 , complex128 , half , uint32 , uint64 . A list of Tensor objects, each with same shape and type. |
shape | A tf.TensorShape or list of ints . Shape of elements of inputs . |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as inputs . |
© 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.3/api_docs/python/tf/raw_ops/AccumulateNV2