def __init__(self, G, seeds):
    self.G = G
    self.seeds = seeds
    self.edges = torch.cat((self.G.edges()[0].unsqueeze(0), self.G.edges()[1].unsqueeze(0)), 0).t()
def __init__(self, G, seeds):
    self.G = G
    self.seeds = seeds
    self.edges = torch.cat((self.G.edges()[0].unsqueeze(0), self.G.edges()[1].unsqueeze(0)), 0).t()