Numba jit и отложенные типы, когда у класса есть переменная, которая является объектом того же класса - PullRequest
Новая
       6

Numba jit и отложенные типы, когда у класса есть переменная, которая является объектом того же класса

0 голосов
/ 20 марта 2020

Этот вопрос основан на сообщении Numba jit и отложенных типах

Это код, указанный в сообщении 

import numpy as np
import numba as nb

spec = [('x', nb.float64)]

@nb.jitclass(spec)
class EarthModel:
    def __init__(self, x):
        self.x = x

earth_model_type = EarthModel.class_type.instance_type

@nb.jit(nb.float64(nb.float64[:, :], nb.float64[:, :], nb.float64[:, :], earth_model_type))
def test(x, y, z, em):
    return em.x

В приведенном выше коде предположим, класс EarthModel имеет переменную, которая является объектом того же класса. Как бы вы тогда указали earth_model_type?

...