tf.keras.layers.Average

TensorFlow 2 version View source on GitHub

Layer that averages a list of inputs.

It takes as input a list of tensors, all of the same shape, and returns a single tensor (also of the same shape).