PyTorch 图神经网络：将 Edge Index 转换为索引列值

要将 edge_index 变为索引列的值，可以使用 range 函数生成以 idx * 20 为起始值，步长为 1 的连续整数序列，并将其赋值给 edge_index 的第一行和第二行。代码如下：

def __getitem__(self, idx):
    edge_index = self.edges.clone()
    edge_index[0] = torch.arange(idx * 20, idx * 20 + edge_index.size(1))
    edge_index[1] = torch.arange(idx * 20, idx * 20 + edge_index.size(1))
    return edge_index

这样，edge_index 的第一行和第二行的值将分别为 idx * 20 到 idx * 20 + edge_index.size(1) 的连续整数序列。