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)