Python 为什么numba中的guvectorize返回奇怪的输出?
例如,我无法理解以下函数的输出:Python 为什么numba中的guvectorize返回奇怪的输出?,python,numpy,numba,Python,Numpy,Numba,例如,我无法理解以下函数的输出: @nb.guvectorize(["void(float64, float64, float64, float64)"], "(),()->(),()") def add_subtract(a, b, res1, res2): res1 = a + b res2 = a - b 当我称之为 add_subtract(np.array([3.0,5.0]), np.array([6.0,7.0]))
@nb.guvectorize(["void(float64, float64, float64, float64)"], "(),()->(),()")
def add_subtract(a, b, res1, res2):
res1 = a + b
res2 = a - b
当我称之为
add_subtract(np.array([3.0,5.0]), np.array([6.0,7.0]))
奇怪的是,它将显示:
(array([6.95307160e-310, 1.23643049e-311]), array([0., 0.]))
此函数似乎试图返回0。为什么函数不返回[[9,-3],[12,-2]]?输出数组未初始化地传递给函数。如果将新值分配给
res1
和res2
局部变量,则将用新数组替换数组。这不会更改传递给函数的数组,这些数组仍处于未初始化状态
相反,您需要替换它们的内容。以下示例有效:
@nb.guvectorize("void(float64[:], float64[:], float64[:], float64[:])",
"(),()->(),()")
def add_subtract(a, b, res1, res2):
res1[:] = a + b
res2[:] = a - b