def __getitem__self idx edge_index = selfedgesclone edge_index0 += idx 20 edge_index1 += idx 20怎么变成索引列的值
要将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)的连续整数序列。
原文地址: https://www.cveoy.top/t/topic/i46d 著作权归作者所有。请勿转载和采集!