嗨,我有以下代码:
@guvectorize(["void(int64[:,:], int64[:,:])"], '(n,m)->(a,b)')
def func (arr, output):
output[1,1] = 0
#more code here to play with output variable as it is of diffent size ans dimension
arr = np.array([ [1,2],[2,3]], dtype='int64')
output = = np.array([[0,0]], dtype='int64')
func(arr, output)
Python提示错误:
在功能签名中无法识别的a,b符号
我的问题是,如何输入n,m维数组并返回x,y或a,b维数组
如果我将a,b重命名为n,m,Numba guvectorize似乎可以工作