python - 为什么 numba 在 (nopython=True) 时抛出有关 numpy 方法的错误？

Question

我正在尝试使用 numba 来提高我编写的一些相当慢的代码的速度。大部分时间花在一个函数上。首先我尝试使用

@jit 

在我的函数定义之前，这稍微改进了计时。然后，我尝试使用

@jit(nopython=True) 

相反。根据我在文档中阅读的内容，应该支持我在函数中使用的 numpy 方法(例如转置)。但是，我收到一个错误

Failed at nopython (nopython frontend)
Untyped global name 'transpose'

Answer 1

为了使用 transpose，您需要以 numpy 数组方法的形式调用它(如文档 describe )。所以以下工作:

import numpy as np
import numba as nb

@nb.jit(nopython=True)
def func(x):
    y = x.transpose()  # or x.T
    return y

x = np.random.normal(size=(4,4))
x_t = func(x)

但是在函数中调用 y = np.transpose(x) 不会。我假设你正在做后者。请注意，我使用的是 Numba 0.25.0 作为引用。