为什么 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 快。