美文网首页
pytorch中的参数类——torch.nn.parameter

pytorch中的参数类——torch.nn.parameter

作者: yuanCruise | 来源:发表于2020-02-10 23:01 被阅读0次
1.torch.nn.parameter概要

pytorch官网对torch.nn.parameter的描述如下。


torch.nn.parameter是一个被用作神经网络模块参数的tensor。这是一种tensor的子类。
parameters是张量的子类,当与模块s一起使用时,它们有一个非常特殊的属性——当它们被重新分配为模块属性时,它们会自动地添加到它的参数列表中,并且会出现在Parameters()迭代器中。分配一个张量没有这样的效果。这是因为可能希望在模型中缓存一些临时状态,比如RNN的最后一个隐藏状态。如果没有Parameter这样的类,这些临时人员也会被注册。

相关文章

网友评论

      本文标题:pytorch中的参数类——torch.nn.parameter

      本文链接:https://www.haomeiwen.com/subject/dtazxhtx.html