gpt4 book ai didi

python - Keras 自定义缩放层

转载 作者:太空宇宙 更新时间:2023-11-04 08:01:08 26 4
gpt4 key购买 nike

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

或者改变合并层,这样就能对输入层做加权平均。谢谢

最佳答案

尝试 Lambda 层

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

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

关于python - Keras 自定义缩放层,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40004706/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com