评价此页

参数#

class torch.nn.parameter.Parameter(data=None, requires_grad=True)[source]#

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

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

参数
  • data (Tensor) – 参数张量。

  • requires_grad (bool, optional) – 参数是否需要梯度。请注意,torch.no_grad() 上下文不会影响 Parameter 创建的默认行为——Parameter 在 no_grad 模式下仍将具有 requires_grad=True。有关更多详细信息,请参见 局部禁用梯度计算。默认值:True