关于with torch.no_grad:的一些小问题

with torch.no_grad:是截断梯度记录的,新生成的数据的都不记录梯度,但是今天产生了一点小疑惑,如果存在多层函数嵌入,是不是函数内所有的数据都不记录梯度,验证了一下,确实是的。

python 复制代码
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)
def add(x,y,z):
    w = x + y + z
    print(w.requires_grad)
    print(w.grad_fn)
def add2(x,y,z):
    add(x,y,z)
with torch.no_grad():
    add2(x,y,z)
    
add2(x,y,z)
"""
输出:
False
None
True
<AddBackward0 object at 0x00000250371BED68>
"""
相关推荐
小钻风33666 分钟前
软件测试: 从入门到实践(接口自动化)
软件测试·python·自动化
别多香了16 分钟前
Python 基础--循环判断&字符串
开发语言·python
老歌老听老掉牙18 分钟前
使用 Matplotlib 自定义坐标轴字体及刻度样式详解
python·matplotlib
股朋公式网1 小时前
斩仙飞刀、 通达信飞刀 源码
python·算法
不吃橘子的橘猫1 小时前
NVIDIA DLI 《Build a Deep Research Agent》学习笔记
开发语言·数据库·笔记·python·学习·算法·ai
薛不痒1 小时前
深度学习介绍以及深度学习相关配置
人工智能·深度学习
学Linux的语莫1 小时前
python的基础使用
开发语言·python
万粉变现经纪人1 小时前
如何解决 pip install SSL 报错 ValueError: check_hostname requires server_hostname 问题
网络·python·网络协议·beautifulsoup·bug·ssl·pip
逻极1 小时前
FastAPI + SQLAlchemy 现代API项目实战:从零到上手的Python MySQL开发指南
python·mysql·fastapi·异步·sqlalchemy
吃人陈乐游刘1 小时前
06实战经验X-anylabelingAI自动标注数据集-本地实现-方法二(2025年12月)保姆级教程
python·miniforge·xanylabeling