参数 ¶

class torch.nn.parameter.Parameter(data=None, requires_grad=True)

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

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

参数:

• data (Tensor) – 参数张量。

• requires_grad (bool, optional) – 如果参数需要梯度。注意，torch.no_grad() 上下文不会影响 Parameter 创建的默认行为 - 在 no_grad 模式下，Parameter 仍然具有 requires_grad=True。有关详细信息，请参阅局部禁用梯度计算。默认：True