#with torch.no_grad() 详解
'''
被该语句包裹起来的语句将不会被追踪梯度
'''
a=torch.tensor([1.1],requires_grad=True)
b=a*2
print(b)#tensor([2.2000], grad_fn=<MulBackward0>)
b.add_(2)
print(b)#tensor([4.2000], grad_fn=<AddBackward0>)
with torch.no_grad():
b.mul_(2)
print(b)#tensor([8.4000], grad_fn=<AddBackward0>)
#被包裹的情况下,b.grad_fn还是add,说明mul并没有被追踪
with torch.no_grad()
最新推荐文章于 2024-06-06 10:19:17 发布
关键词由CSDN通过智能技术生成