为什么 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 作为参考。

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

Why is numba throwing an error regarding numpy methods when (nopython=True)?

python

numpy

numba