在pytorch中创建可学习参数时出现问题
使用torch.tensor()
创建参数,并令requires_grad=true
self.rho = torch.tensor(0.5, requires_grad=True)
这样创建的参数是可以学习的参数 (有梯度)
但是并没有绑定在module的parameter列表中!!!!!也就是说不能作为参数保存下来!!!!
所以要用nn.Parameter()
以下是修正后的代码
self.rho = nn.Parameter(torch.tensor(0.5), requires_grad=True)