是否可以定义递归数据类型的numba jitclass

时间:2020-06-24 11:17:19

标签: numba recursive-datastructures

我想加快一个类似树形数据结构的简单实现的python代码。首先想到的是将其与numba同步。我遇到的问题是如何告诉numba Node的类型,因为它的成员之一又是Node类型。这是一个最小的示例:

spec=[("value", nb.float64), ("parent", Node.class_type.instance_type)]
@nb.jitclass(spec)
class Node:
    def __init__(self, value:float, parent:"Node") -> None:
        self.value:float = value
        self.parent:Node = parent

该示例显然不起作用,因为在定义spec时,尚不知道如何处理类Node

在numba中甚至可能吗?

0 个答案:

没有答案
相关问题