参考 PyTorch里面的torch.nn.Parameter()详解 - 云+社区 - 腾讯云
class torch.nn.parameter.
Parameter
[source]
作为一种模型参数的张量。参数是张量的子类,在与模块s一起使用时具有一个非常特殊的特性——当它们被指定为模块属性时,它们会自动添加到参数列表中,并且会出现,例如在parameters()迭代器中。
分配一个张量就没有这样的效果。这是因为人们可能想要在模型中缓存一些临时状态,比如RNN的最后一个隐藏状态。如果没有像Parameter的类,这些临时变量也要注册。
参数:
-
data (Tensor) – 参数张量
-
requires_grad (bool, optional) – 参数是否需要梯度. 看Excluding subgraphs from backward上更多细节。 默认为真。