Numba签名协议

Question

尽管搜索堆栈溢出和整个互联网，阅读几个堆栈溢出问题和numba.pydata.org页面，并学习一些线索来告诉Numba我想给Numba什么类型，并从函数中得到什么，但我并没有找到它工作的实际逻辑。

例如，我试验了一个函数，该函数处理整数列表并输出另一个整数列表，当装饰器@numba.jit(numba.int64[:](numba.int64[:]))工作时，装饰器@numba.njit(numba.int64[:](numba.int64[:]))和@numba.vectorize(numba.int64[:](numba.int64[:]))不能工作。

(njit成功地通过了修饰器，无意中发现了函数本身；我猜在'no python'模式下，将元素连接到list并不是可用的函数。然而，vectorize抱怨签名，TypeError: 'Signature' object is not iterable；也许它担心1D数组可能包含一个没有方括号的元素，而这个元素是不可迭代的？)

是否有一个简单的方法来理解Numba如何工作到足够的深度来预测我应该如何表达签名？

Answer 1

对于jit (以及njit，也就是nopython=True的别名)来说，最简单的答案是尽量避免编写签名--在常见情况下，类型推断会使您达到目的。

针对您的问题，numba.int64[:](numba.int64[:])是一个有效的签名，适用于jit。

numba.vectorize-期望一个可迭代的签名(错误消息)，所以您的签名需要包装在一个列表中。另外，vectorize创建了一个numpy ufunc，它由标量操作(然后广播)定义，因此您的签名必须是标量类型。例如。

@numba.vectorize([numba.int64(numba.int64)])
def add_one(v):
    return v + 1

add_one(np.array([4, 5, 6], dtype=np.int64))
# Out[117]: array([5, 6, 7], dtype=int64)