用代码举例pytorch的自动求导

下面是一个简单的例子，用 PyTorch 的自动求导功能计算 y = x^2 在 x = 3 处的导数：

import torch

# 创建一个张量 x，值为 3
x = torch.tensor(3.0, requires_grad=True)

# 计算 y = x^2
y = x**2

# 对 y 进行反向传播，自动求导
y.backward()

# 输出 x 在 x = 3 处的导数
print(x.grad)

输出结果为：

tensor(6.)

说明 x 在 x = 3 处的导数为 6。在上面的代码中，我们创建了一个张量 x，并将 requires_grad 参数设置为 True，表示需要对它进行自动求导。然后，我们计算了 y = x^2，并对 y 进行了反向传播，自动求导。最后，我们输出了 x 在 x = 3 处的导数，即 y 对 x 的导数。