Python中的Numba Guvectorize签名错误

时间:2019-01-01 16:16:24

标签: python numba gpu-programming

嗨,我有以下代码:

@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似乎可以工作

0 个答案:

没有答案