Keras自定义缩放层

Question

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

或者更改合并图层，它将能够对输入图层进行加权平均。 感谢

Answer 1

尝试使用Lambda图层

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

https://keras.io/layers/core/#lambda

Answer 2

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

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