计算图
非标量调用backward()求梯度
x = torch.arange(4.0)
x.requires_grad = True
y = x * x
方法1:通过y.sum().backward()调用反向传播
y.sum().backward()
print(x.grad)
本文介绍了深度学习中如何对非标量进行梯度计算,包括两种调用反向传播的方法:通过y.sum().backward()和传入ones参数。同时,解释了detach()函数的作用,它能从计算图中分离张量并阻止其进一步参与反向传播。
x = torch.arange(4.0)
x.requires_grad = True
y = x * x
y.sum().backward()
print(x.grad)
763
1764
2277