with torch.no_grad()

讲述with torch.no_grad()前,先讲述requires_grad参数。

1. requires_grad

torch 中,tensor 有一个 requires_grad参数,如果设置为 True,则反向传播时,该 tensor 就会自动求导。

tensorrequires_grad的属性默认为 False,若一个节点(叶子变量:自己创建的 tensorrequires_grad被设置为True,那么所有依赖它的节点requires_grad都为 True(即使其他相依赖的 tensorrequires_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()

即使一个 tensor (命名为 x)的requires_grad = True,由 x 得到的新 tensor(命名为 wrequires_grad也为 False,且grad_fn也为 None,即不会对 w 求导。

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


posted @   做梦当财神  阅读(235)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 无需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 的用法
点击右上角即可分享
微信分享提示