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

已采取

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

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
今天

您可以使用一个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])

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章