0
点赞
收藏
分享

微信扫一扫

nn.parameter

参考 PyTorch里面的torch.nn.Parameter()详解 - 云+社区 - 腾讯云

class torch.nn.parameter.Parameter[source]

作为一种模型参数的张量。参数是张量的子类,在与模块s一起使用时具有一个非常特殊的特性——当它们被指定为模块属性时,它们会自动添加到参数列表中,并且会出现,例如在parameters()迭代器中。分配一个张量就没有这样的效果。这是因为人们可能想要在模型中缓存一些临时状态,比如RNN的最后一个隐藏状态。如果没有像Parameter的类,这些临时变量也要注册。

参数:

  • data (Tensor) – 参数张量

  • requires_grad (bool, optional) – 参数是否需要梯度. 看Excluding subgraphs from backward上更多细节。 默认为真。

举报

相关推荐

0 条评论