我正在尝试在Python 2.7中创建一个函数,该函数接受函数句柄和函数中的变量数作为输入,并返回一个新函数,该函数计算输入函数的梯度。到目前为止,这就是我所拥有的。
import sympy as sym
import numpy as np
def getSymbolicGradient(func,numVars):
# Initialize Some Variables
g = numVars * [0]
# Create All the Symbolic Variables
x = sym.symarray('x',numVars)
# Calculate the Gradients
for i in range(numVars):
g[i] = sym.diff(func(x),x[i])
gradFunc = sym.lambdify(x, g, modules="numpy")
return gradFunc
说我使用gradFunc
以下代码:
def myVecFunc(x):
return 2*x[0]**2 + 4*x[1] + 2
gradFunc = getSymbolicGradient(func=myVecFunc, numVars=2)
如果我使用两个单独的参数调用它,则它可以正常工作,例如:
print( gradFunc(1,2) )
但是,如果我使用单个参数(例如Numpy数组)来调用它,
print( gradFunc(np.array([1,2])) )
我收到以下错误:
TypeError:()正好接受2个参数(给定1个)
如何获得lambdify
输入参数作为单个值内部的单个数组?是否有更好的(内置)Sympy方法来为将数组作为输入的函数的梯度生成符号表达式?
我对numpy不太熟悉,但是通常在Python中,您可以使用*
运算符解压缩数组值。
a = [2, 4, 6]
my_func(*a)
在逻辑上等同于
a = 2
b = 4
c = 6
my_func(a, b, c)
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句