标量函数对向量求导的python代码实现

假设标量函数为 f(x0,x1) = x0 **2 + x1 **2
这里采用的是数值微分，不是精确地解析微分，但是准确率也不差。
直接上源代码

import numpy as np
def function_2(x):     # x数组中只有两个数，假设标量函数为 f(x0,x1) = x0 **2 + x1 **2
    return x[0]**2 + x[1]**