使用Keras层时的TypeError

Question

我正在学习Keras并尝试构建我的层，所以我构建了一个简单的Lambda层，它只是计算输入的平均值。

当我将该层添加到顺序模型时，TypeError发生在第4行。这是我的代码：

mean_layer = keras.layers.Lambda(lambda x: tf.reduce_mean(x))
model = keras.models.Sequential()
model.add(keras.layers.InputLayer(input_shape=[10]))
model.add(mean_layer())
model.summary()

TypeError: __call__() missing 1 required positional argument: 'inputs'

然而，如果我使用函数模型，事情进行得很好，我得到了预期的输出。

input = keras.layers.Input(shape=[10])
output = mean_layer(input)
model = keras.models.Model(inputs=[input], outputs=[output])
model.summary()

当我使用顺序API时，我是否以错误的方式构建了模型？谢谢

Answer 1

是的，你做错了，应该是这样的：

model.add(mean_layer)