Keras 2：在“合并”层中使用lambda函数

Question

我正在尝试实现此合并层：

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

但是，在Keras 2中不再存在“合并”。您只能访问经过公共标准化的“合并”层，例如“加”，“乘”，“点”。

如何在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])