Làm cách nào để sử dụng numba trên một hàm thành viên của một lớp?

Question

Tôi đang sử dụng phiên bản ổn định của Numba 0.30.1.

tôi có thể làm điều này:

import numba as nb 
@nb.jit("void(f8[:])",nopython=True)        
def complicated(x):         
    for a in x: 
     b = a**2.+a**3. 

như một trường hợp thử nghiệm, và tăng tốc là rất lớn. Nhưng tôi không biết làm thế nào để tiến hành nếu tôi cần tăng tốc một hàm bên trong một lớp.

import numba as nb 
def myClass(object): 
    def __init__(self): 
     self.k = 1 
    #@nb.jit(???,nopython=True)        
    def complicated(self,x):         
     for a in x: 
      b = a**2.+a**3.+self.k 

Tôi nên sử dụng loại numba nào cho đối tượng self? Tôi cần phải có hàm này bên trong một lớp vì nó cần truy cập vào biến thành viên.

Answer 1

Bạn có một vài lựa chọn:

Sử dụng một jitclass (http://numba.pydata.org/numba-doc/0.30.1/user/jitclass.html) để "numba-ize" toàn bộ điều.

Hoặc thực hiện chức năng thành viên một wrapper và vượt qua các biến thành viên thông qua:

import numba as nb 

@nb.jit 
def _complicated(x, k): 
    for a in x: 
     b = a**2.+a**3.+k 

def myClass(object): 
    def __init__(self): 
     self.k = 1 

    def complicated(self,x):         
     _complicated(x, self.k)