爲什麼numba在numpy的linspace中引發一個類型錯誤

Question

我正在使用numba 0.34.0和numpy 1.13.1。一個小例子被示出如下：

import numpy as np  
from numba import jit 
@jit(nopython=True) 
def initial(veh_size): 
    t = np.linspace(0, (100 - 1) * 30, 100, dtype=np.int32) 
    t0 = np.linspace(0, (veh_size - 1) * 30, veh_size, dtype=np.int32) 
    return t0 

initial(100) 

兩者符合t和t0具有相同的錯誤消息。

錯誤消息：

numba.errors.InternalError: 
[1] During: resolving callee type: Function(<function linspace at 0x000001F977678C80>) 
[2] During: typing of call at ***/test1.py (6) 

Answer 1

由於np.linspace的numba版本不接受任何dtype參數（source: numba 0.34 documentation）：

2.7.3.3。其它功能

以下頂層功能被支持：

• [...]

• numpy.linspace()（僅3參數形式）

• [... ]

您需要使用astype將其轉換nopython-numba函數中：

import numpy as np  
from numba import jit 
@jit(nopython=True) 
def initial(veh_size): 
    t = np.linspace(0, (100 - 1) * 30, 100).astype(np.int32) 
    t0 = np.linspace(0, (veh_size - 1) * 30, veh_size).astype(np.int32) 
    return t0 

initial(100) 

或者只是不nopython-numba函數使用np.linspace並把它作爲參數。這避免了一個臨時陣列，我懷疑Numbas np.linspace比NumPys更快。