tf.keras.layers.Average

TensorFlow 1 version View source on GitHub

Layer that averages a list of inputs.

tf.keras.layers.Average(
    **kwargs
)

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