参数

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

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

参数是 Tensor 的子类，在与 Module 一起使用时具有特殊属性：当它们作为模块的属性分配时，会自动添加到该模块的参数列表中，并在例如 parameters() 迭代器中显示。而普通的 Tensor 赋值则不会产生这样的效果。这是因为人们可能希望在模型中缓存一些临时状态，比如 RNN 的最后一个隐藏状态。如果没有 Parameter 类，这些临时变量也会被注册。

参数

• data (Tensor) – 表示参数的张量。

• requires_grad (bool, 可选) – 表示参数是否需要计算梯度。注意，torch.no_grad() 上下文不会影响 Parameter 创建的默认行为，在 no_grad 模式下，默认情况下 requires_grad 仍然为True。更多详情请参阅局部禁用梯度计算。默认值： True