python - 使用 Keras Lambda 层时出现类型错误

Question

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

当我将此层添加到顺序模型时，第 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)