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

问题描述 投票:1回答:1

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

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

但是,Keras 2中不再出现“merge”。您只能访问公共标准化的“Merge”图层,例如Add,Multiply,Dot。

如何在Keras 2中实现此功能?我想过制作两个合并图层,但我不知道如何实现它,特别是因为“K.mean”部分。

供参考,以下是进口:

from keras.layers import merge
from keras import backend as K
python keras keras-layer keras-2
1个回答
2
投票

您可以使用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])
© www.soinside.com 2019 - 2024. All rights reserved.