def sample(self, seeds):
    ' seeds torch.LongTensor : a batch of indices of edges '
    return self.edges[torch.LongTensor(seeds)]
seeds torch.LongTensor : a batch of indices of edges
def sample(self, seeds):
    '  '
    return self.edges[torch.LongTensor(seeds)]