Parameter

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

一个被视为模块参数的 Tensor 类型。

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

参数

• data (Tensor) – 参数张量。

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