在课堂上使用 numba？

Question

我可以在 0.6 numba 文档中找到一些关于如何使用 numba on classes 的信息：

from numba import jit, void, int_, double

# All methods must be given signatures

@jit
class Shrubbery(object):
    @void(int_, int_)
    def __init__(self, w, h):
        # All instance attributes must be defined in the initializer
        self.width = w
        self.height = h

        # Types can be explicitly specified through casts
        self.some_attr = double(1.0)

    @int_()
    def area(self):
        return self.width * self.height

    @void()
    def describe(self):
        print("This shrubbery is ", self.width,
              "by", self.height, "cubits.")

但我在0.16 documentation 中找不到。是否总是可以在类上使用 numba ？

Answer 1

从 0.23 版开始，有一个 numba.jitclass 方法。我可以说以下适用于 0.26 版

@numba.jitclass([('width', numba.float64), ('height', numba.float64)])
class Shrubbery(object):
    def __init__(self, w, h):
        self.width = w
        self.height = h

    def area(self):
        return self.width * self.height

shrubbery = Shrubbery(10, 20)
print(shrubbery.area())

documentation 表示只允许使用方法和属性，因此很遗憾，包含 describe 函数目前将不起作用。

Answer 2

我最后一次听说numba class support是他们temporarily removed it since 0.12