我是新来Pytorch和我感到困惑的区别nn.Parameter 和autograd.Variable。我知道前一个是 的子类Variable并且具有梯度。但是我真的不明白我们为什么要引入Parameter,什么时候应该使用它?
总结:
感谢iacolippo的解释,我终于明白了parameter和的区别variable。总而言之,variablepytorch 中的variable与 tensorflow 中的不同,前者不附加到模型的可训练参数,而后者会。附加到模型意味着使用model.parameters()将返回某个参数给你,这在训练阶段指定训练所需的变量很有用。“变量”作为某些网络中的缓存更有帮助。