Loading...

正文

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

含义是:通过将一个固定不可训练的张量(Tensor)转换成可训练的类型Parameter,并将这个Parameter绑定到模块(Module)中(即在net.parameters()中可以找到这个绑定的Parameter),从而在参数优化时能够对其进行优化。因此,通过这种类型转换,self.v变成了模型的一部分,成为了模型中可以根据训练进行调整的参数。使用这个函数的目的是为了让某些变量在学习过程中能够不断地修改其值,以达到最优化。


参数

  • data (Tensor) - 参数张量。
  • requires_grad (bool, 可选) - 如果参数需要梯度。注意,torch.no_grad()上下文不影响Parameter创建的默认行为——即使在no_grad模式下,Parameter仍将有requires_grad=True。更多详情见局部禁用梯度计算。默认值:True


参考链接