PyTorch如何一次性计算多个节点的梯度?

JinxMiracle 2017-07-14 04:18:47
本人新手,刚刚学PyTorch,在通过backward计算梯度时遇到了一些问题:

可以看见,能够计算x的梯度,但是y的梯度却为none,这是为什么呢?难道计算的时候,只能计算末尾节点的梯度么?
...全文
350 2 打赏 收藏 转发到动态 举报
写回复
用AI写文章
2 条回复
切换为时间正序
请发表友善的回复…
发表回复
sky_zdk 2018-11-26
  • 打赏
  • 举报
回复
因为x设置requires_grad=True 所以在计算梯度时候,只算这个tensor的梯度,而其他tensor(默认false),则不会计算梯度,减少内部的计算量
JinxMiracle 2017-07-15
  • 打赏
  • 举报
回复
有人来帮帮忙么?

37,719

社区成员

发帖
与我相关
我的任务
社区描述
JavaScript,VBScript,AngleScript,ActionScript,Shell,Perl,Ruby,Lua,Tcl,Scala,MaxScript 等脚本语言交流。
社区管理员
  • 脚本语言(Perl/Python)社区
  • IT.BOB
加入社区
  • 近7日
  • 近30日
  • 至今

试试用AI创作助手写篇文章吧