b1 = nn.Parameter(torch.zeros(num_hiddens, requires_grad=True))中为什么要用(torch.zeros
时间: 2024-05-29 18:13:35 浏览: 156
浅谈pytorch中torch.max和F.softmax函数的维度解释
5星 · 资源好评率100%
(num_hiddens, requires_grad=True))创建一个nn.Parameter对象?
这是因为我们需要对这个参数进行训练。而使用nn.Parameter可以将其注册为模型参数,使得该参数能够被自动优化器所更新。torch.zeros则是创建了一个大小为num_hiddens的零向量,作为b1的初始值。requires_grad=True则是使得该参数在反向传播时可以被求导。
阅读全文