设置一个可学习参数

最新推荐文章于 2024-04-11 09:57:52 发布

QT-Smile

最新推荐文章于 2024-04-11 09:57:52 发布

阅读量274

点赞数

分类专栏：深度学习之美文章标签：学习深度学习 python

本文链接：https://blog.csdn.net/qq_34848334/article/details/129458734

版权

深度学习之美专栏收录该内容

209 篇文章 24 订阅

订阅专栏

设置一个可学习参数

#设置前置网络及 可学习参数
self.cnn=cnn_output4()
self.fuse_weight_1 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)
self.fuse_weight_2 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)
self.fuse_weight_3 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)
self.fuse_weight_4 = torch.nn.Parameter(torch.FloatTensor(1), requires_grad=True)
 
#初始化
self.fuse_weight_1.data.fill_(0.25)
self.fuse_weight_2.data.fill_(0.25)
self.fuse_weight_3.data.fill_(0.25)
self.fuse_weight_4.data.fill_(0.25)
 
def forward(x):
    x1,x2,x3,x4=self.cnn(x)
    return fuse_weight_1*x1+fuse_weight_2*x2+fuse_weight_3*x3+fuse_weight_4*x4