如何以模式作为任何用户定义的函数来实现keras的合并层

Question

我正在尝试撰写有关CNN专家的论文，我需要添加合并层作为每个CNN输出的加权总和，其中权重是选通网络的输出。所以我的模式是一个自定义函数。

这是我要运行的代码：

def merge_mode(branches):
    g, o1, o2, o3, o4 = branches

    return K.transpose(K.transpose(o1)*g[:,0] + K.transpose(o2)*g[:,1] +             K.transpose(o3)*g[:,2] + K.transpose(o4)*g[:,3])

model = Sequential()
model.add(merge([gate, model1, model2, model3, model4], output_shape=(3,), mode=merge_mode))