如何在 Numba 中使用 Python 类

Question

我有 Numba 0.24，它支持类。

当我尝试构建最简单的类时，我可以想象我发现了一个错误！发生了什么事？

from numba import jitclass
@jitclass
class foo:
    x = 2
bar = foo()

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-2-3e0fd8d4bd2b> in <module>()
      3 class foo:
      4     x = 2
----> 5 bar = foo()

TypeError: wrap() missing 1 required positional argument: 'cls'

我错过了什么吗？

Answer 1

您需要指定规格：

spec = [('x', nb.int64)]
@nb.jitclass(spec)
class foo(object):
    def __init__(self):
        self.x = 2

bar = foo()
print bar.x

看看docs。此时不支持类变量。你必须使用实例变量。