3.2 Linear regression

被with torch.no_grad()包住的代码,不用跟踪反向梯度计算,来做一个实验:

a = torch.tensor([1.1], requires_grad=True)
b = a * 2

输出是:

tensor([2.2000], grad_fn=) 可以看到梯度函数是MulBackward0,表示乘法的反向梯度函数

b.add_(2)

输出是:

tensor([4.2000], grad_fn=) 表明是add的反向梯度函数

with torch.no_grad():
    b.mul_(2)

结果是:

tensor([8.4000], grad_fn=) 可以看到没有跟踪乘法的梯度,还是上面的加法的梯度函数,不过乘法是执行了的

你可能感兴趣的:(线性回归,深度学习)