Pytorch的自动微分---手动微分

# 定义一个函数f(x)=x^2+3*x+2
import matplotlib.pyplot as plt
import numpy as np


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


# 手动求导
def df(x):
    return 2 * x + 3


# 在main函数中，使用linspace，生成自变量x的序列
if __name__ == '__main__':
    x = np.linspace(-6.5, 3.5, 1000)
    # 计算函数值
    y_f = f(x)
    # 计算导函数值
    y_df = df(x)
    # 绘制函数图像和导函数图像
    plt.plot(x, y_f, label='f(x)=x*x+3*x+2')
    plt.plot(x, y_df, label="f'(x)=2*x+3")
    plt.legend()
    plt.grid(True)
    plt.show()