如何在keras 2.0中使用带有mul运算符的合并

Question

我有keras 1.0的代码

from keras.layers import  merge

predict_vector = merge([a, b], mode = 'mul')

由于merge不见了，那么喀拉斯语merge中2.0以上的等效代码是什么？

Answer 1

您可以尝试lambda-layer，它可以满足您的要求。这是一个例子

import keras.backend as K

def mulFun(x):
    return x[0] * x[1]
first = Dense( input_shape1 )(x1)
second = Dense( input_shape2 )(x2)
# perform  opration
Lambda(mulFun, output_shape)([first, second])

这里another-good-example正在使用。