Graph 火炬传递时出错

我运行了一个在github中找到的项目，并在“IndexError:Dimension超出范围（预期在[-1,0]范围内，但得到了-2）”时出错。

此错误与torch_geometric.nn模块中的类messagepassing（）有关

源代码来自

关于自我。设置大小（）功能：

def __set_size__(self, size: List[Optional[int]], dim: int, src: Tensor):
    the_size = size[dim]
    if the_size is None:
        size[dim] = src.size(self.node_dim)
    elif the_size != src.size(self.node_dim):
        raise ValueError(
            (f'Encountered tensor with size {src.size(self.node_dim)} in '
             f'dimension {self.node_dim}, but expected size {the_size}.'))

我不确定这是否是数据的问题。如果你知道这种情况出了什么问题，请帮助我

def __set_size__(self, size: List[Optional[int]], dim: int, src: Tensor):
    the_size = size[dim]
    if the_size is None:
        size[dim] = src.size(self.node_dim)
    elif the_size != src.size(self.node_dim):
        raise ValueError(
            (f'Encountered tensor with size {src.size(self.node_dim)} in '
             f'dimension {self.node_dim}, but expected size {the_size}.'))