pytorch自动求梯度autograd的实现

autograd是pytorch构建神经网络的核心。

在 PyTorch 中，结合以下代码例子，当你对一个张量 x 设置 requires_grad=True 时，你China编程实际上是告诉 PyTorch 需要跟踪所有对这个张量的操作以便未来可以计算梯度。当你调用 out.backward() 时，PyTorch 会自动计算关于 out 的梯度，并逆向传播回 x。

在以下代码中：

x = torch.ojavascriptnes(2,2)
print(x)
x.requires_grad=True
print(x)
y = x + 2
z = y*y*3
out = z.mean()
print(out)
out.backward()
print(x.grad)
print(y.grad)  # 这一行会报错，因为 y 不需要梯度，y为非叶子张量

y 是由 x 通过一个简单的加法操作得到的。由于 y 本身并没有设置 requires_grad=True，PyTorch 没有跟踪 y 的操作历史，因此无法计算 y 关于 out 的梯度。
如果你想要计算 y 的梯度，你需要在创建 y 之前也设置 requires_grad=True。但是，通常情况下，我们不需要对中间变量求梯度，我们只需要对最终输出的梯度感兴趣，然后通过反向传播计算出所有需要的梯度。

如果你确实需要对 y 求梯度，可以这样做：

x = torch.ones(2,2, requires_grad=True)
y = x + 2
z = y*y*3
out = z.mean()
out.backward()
prphpint(x.grad)  # 计算 x 的梯度
print(y.grad)  # 现在可以计算 y 的梯度了

在这个修改后的代码中，x 和 y 都设置了 requires_grad=True，因此 PyTorch 会跟踪它们所有的操作，你可以计算 y 关于 out 的梯度。但是请注意，通常情况下，我们只需要计算最终输出（在这个例子中是 out）关于输入变量（在这个例子中是 x）的梯度。

到此这篇关于pytorch自动求梯度autogrFJntqad的实现的文章就介绍到这了,更多相关pytorch自动求梯度autograd内容请搜索编程China编程(www.cppcns.cojsm)以前的文章或继续浏览下面的相关文章希望大家以后多多支持China编程(www.chinasem.cn)！