I am getting a tuple of tensors in the pos_sample given in Node2Vec.
    def pos_sample(self, batch):
        batch = batch.repeat(self.walks_per_node)
        rowptr, col, _ = self.adj.csr()
        rw = random_walk(rowptr, col, batch, self.walk_length, self.p, self.q) 
        # rw is a tuple of tensors 
        walks = []
        num_walks_per_rw = 1 + self.walk_length + 1 - self.context_size
        for j in range(num_walks_per_rw):
            walks.append(rw[:,j:j + self.context_size])
        return torch.cat(walks, dim=0)

Any Idea on how to fix this?