pytorch 画导数

要使用PyTorch绘制导数曲线，你可以遵循以下步骤：

1. 导入必要的库：

import torch
import torch.autograd as autograd
import matplotlib.pyplot as plt

2. 创建一个函数，其中包含你想要绘制导数的数学表达式：

def f(x):
    return x**2 + 2*x + 1

3. 创建一个输入张量，并使用`requires_grad=True`来指示需要计算梯度：

x = torch.linspace(-5, 5, 100, requires_grad=True)

4. 计算函数关于输入的导数：

y = f(x)
grads = autograd.grad(y, x, create_graph=True)[0]

5. 绘制函数和导数曲线：

plt.plot(x.detach().numpy(), y.detach().numpy(), label='function')
plt.plot(x.detach().numpy(), grads.detach().numpy(), label='gradient')
plt.legend()
plt.show()

这些步骤将计算并绘制函数以及它的导数曲线。请注意，我们使用`.detach().numpy()`将张量转换为NumPy数组，以便将其传递给Matplotlib进行绘制。