Keras自定义缩放层

Question

对于我的工作，我需要创建一个只有单一权重的层，它将当前层中的数据乘以一些训练值。有办法做到这一点吗?

或者改变合并层，这将能够对输入层进行加权平均。由于

Answer 1

尝试Lambda图层

model.add(Lambda(lambda x: x *MyValue))

https://keras.io/layers/core/λ

Answer 2

扩展@user375348的答案，您可以创建一个学习变量并在Lambda中使用它:

initial_value = 1
learned = tf.Variable(initial_value, name='learned_scalar')
Lambda(lambda x: x * learned)