Keras 3 API 文档 / 层 API / 合并层 / 平均层

平均层

[源代码]

Average

keras.layers.Average(**kwargs)

按元素平均多个输入。

它接收一个张量列表作为输入,所有张量都具有相同的形状,并返回一个单个张量(也具有相同的形状)。

示例

>>> input_shape = (2, 3, 4)
>>> x1 = np.random.rand(*input_shape)
>>> x2 = np.random.rand(*input_shape)
>>> y = keras.layers.Average()([x1, x2])

在 Keras 模型中的用法

>>> input1 = keras.layers.Input(shape=(16,))
>>> x1 = keras.layers.Dense(8, activation='relu')(input1)
>>> input2 = keras.layers.Input(shape=(32,))
>>> x2 = keras.layers.Dense(8, activation='relu')(input2)
>>> # equivalent to `y = keras.layers.average([x1, x2])`
>>> y = keras.layers.Average()([x1, x2])
>>> out = keras.layers.Dense(4)(y)
>>> model = keras.models.Model(inputs=[input1, input2], outputs=out)