with torch.no_grad()
讲述with torch.no_grad()
前,先讲述requires_grad
参数。
1. requires_grad
在 中, 有一个 requires_grad
参数,如果设置为 ,则反向传播时,该 就会自动求导。
的requires_grad
的属性默认为 ,若一个节点(叶子变量:自己创建的 )requires_grad
被设置为,那么所有依赖它的节点requires_grad
都为 (即使其他相依赖的 的 requires_grad = False
)。
import torch
x = torch.randn(10, 5, requires_grad = True)
y = torch.randn(10, 5, requires_grad = False)
z = torch.randn(10, 5, requires_grad = False)
w = x + y + z
print(w.requires_grad)
True
2. with torch.no_grad()
即使一个 (命名为 )的requires_grad = True
,由 得到的新 (命名为 )requires_grad
也为 ,且grad_fn
也为 ,即不会对 求导。
import torch
x = torch.randn(10, 5, requires_grad = True)
y = torch.randn(10, 5, requires_grad = True)
z = torch.randn(10, 5, requires_grad = True)
with torch.no_grad():
w = x + y + z
print(w.requires_grad)
print(w.grad_fn)
print(w.requires_grad)
False
None
False
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· C#/.NET/.NET Core优秀项目和框架2025年2月简报
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· 【杭电多校比赛记录】2025“钉耙编程”中国大学生算法设计春季联赛(1)
2019-03-06 Jupyter Notebook插入图片的4种方法
2018-03-06 Python with as 的用法