在numba jitclass中声明变量

时间:2016-11-09 21:52:41

标签: python numba

当我尝试导入以下jitclass时,我在字典上遇到KeyError:

from numba import jitclass, float64

spec = [('n', float64),
        ('w', float64),
        ('a', float64)]

@jitclass(spec)
class foo:

    def __init__(self,n,w):

        self.n = n
        self.w = w

    def foo2(self):

        a = self.n*self.w

        return a + 1.

我不明白错误的来源。 numba文档不足以解释事物。我是否需要所有变量都是类变量?

1 个答案:

答案 0 :(得分:1)

65需要继承jitclass

object

这似乎没有在文档中作为要求提及,但它是如何编写示例的。