参数¶

class torch.nn.parameter.Parameter(data=None, requires_grad=True)[源代码]¶

一种被视为模块参数的张量。

参数是 Tensor 的子类，当与 Module 一起使用时，它们具有一个非常特殊的属性——当它们被分配为模块属性时，它们会自动添加到模块的参数列表中，并且会出现在例如 parameters() 迭代器中。分配一个 Tensor 不会产生这样的效果。这是因为可能希望在模型中缓存一些临时状态，比如 RNN 的最后一个隐藏状态。如果没有像 Parameter 这样的类，这些临时状态也会被注册。