Keras 2：在 "Merge" 层中使用 lambda 函数

Question

我正在尝试实现这个合并层：

policy = merge([out1, out2], mode = lambda x: x[0]-K.mean(x[0])+x[1], output_shape = (out_node,))

但是，"merge" 不再出现在 Keras 2 中。您只能访问 public 个标准化的 "Merge" 层，例如添加、乘法、点。

如何在 Keras 2 中实现此功能？我考虑过制作两个合并层，但我不知道如何实现它，尤其是因为 "K.mean" 部分。

作为参考，这里是导入：

from keras.layers import merge
from keras import backend as K

Answer 1

您可以使用 Lambda 层简单地执行此操作：

from keras import backend as K
from keras.layers import Lambda

policy = Lambda(lambda x: x[0] - K.mean(x[0]) + x[1])([out1, out2])

Keras 2: Using lambda function in "Merge" layers