Autograd是PyTorch的自动求导引擎。它可以自动计算张量的梯度,并且可以为任何计算图提供快速的反向传播。以下是使用Autograd的代码示例:
import torch
x = torch.tensor(2.0, requires_grad=True) y = torch.tensor(4.0, requires_grad=True)
z = x3 + y2
z.backward()
print(x.grad) # tensor(12.) print(y.grad) # tensor(8.)