我正在try 使用jit来编译我的代码.目前，除了Numba之外，我唯一使用的库是Numpy.当指定nopPython = True时，它会引发一系列linSpace数组错误.我用一个非常简单的函数重新创建了错误:
@jit(nopython=True)
def func(Nt):
time = np.linspace(np.int_(0),np.int_(Nt-1),np.int_(Nt),dtype=np.int_)
return time
Nt = 10
func(Nt)
运行时，会显示以下错误消息(请参阅随附图像).
我try 过许多不同的排列来扰乱论点的类型，但没有成功.我所要做的就是创建一个从0到Nt - 1的integer的linSpacearray.有什么建议吗？
删除dtype=参数:
import numba
@numba.jit(nopython=True)
def func(Nt):
time = np.linspace(np.int_(0), np.int_(Nt - 1), np.int_(Nt))
return time
Nt = 10
print(func(Nt))
输出:
[0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]
如果您想要的是integer，只需稍后重新铸造即可:
@numba.jit(nopython=True)
def func(Nt):
time = np.linspace(np.int_(0), np.int_(Nt - 1), np.int_(Nt)).astype("int_")
return time