tensor.detach()
x = torch.tensor(2.0)
x.requires_grad_(True)
y = 2 * x
z = 5 * x
w = y + z.detach()
w.backward()
print(x.grad)
=> 2
本来应该x的梯度为7,但是detach()那一路切段了梯度的传播,导致5没有向后传递
x = torch.tensor(2.0)
x.requires_grad_(True)
y = 2 * x
z = 5 * x
w = y + z.detach()
w.backward()
print(x.grad)
=> 2
本来应该x的梯度为7,但是detach()那一路切段了梯度的传播,导致5没有向后传递
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
· 上周热点回顾(3.3-3.9)
· winform 绘制太阳,地球,月球 运作规律