def partial_derivative(f, input):
    ret = np.empty(len(input))
for i in range(len(input)):
    fg = lambda x: partial_function(f, input, i, x)
ret[i] = nd.Derivative(fg)(input[i])
return ret
if __name__ == '__main__':
    f = lambda x, y: x * x * x + y * y
input = np.array([1.0, 1.0])
print('partial_derivative of f() at: ' + str(input))
print(partial_derivative(f, input))
Векторизованная частная производная многомерной функции в Python