首页 > 解决方案 > 在 keras >2.0 中使用什么来代替 mode='mul' 的合并?

问题描述

我正在尝试修改一些为 Keras<2.0 编写的代码,但我偶然发现了这行代码:

output_attention_mul = merge([inputs, a_probs], name='attention_mul', mode='mul')

如何在 Keras>2.0 中翻译这行代码?

我试过以Multiply这种方式使用图层

output_attention_mul = Multiply()([inputs, a_probs])

但我真的不知道这是否是正确的做法,因为在旧代码中它使用的是merge操作而不是Merge层。

如果我做了正确的事情或者我使用了错误的方法,任何人都可以确认我吗?我应该使用multiply而不是Multiply吗?

标签: pythonkeras

解决方案


我认为你正在寻找的是keras.layers.Multiply()

那应该给你output_attention_mul = keras.layers.Multiply()([inputs, a_probs])

您应该可以在此处的文档中找到:https ://keras.io/layers/merge/

希望这对你有帮助!


推荐阅读