这个错误是由于 GCN 网络中的边索引超出了范围引起的。根据提供的边连接关系,可以看出边索引的最大值是 36,但是在 GCN 网络中,边索引的维度应该是节点的数量,即 61。这导致了超出维度范围的错误。

要解决这个问题,你需要将边索引调整为正确的维度。根据提供的边连接关系,你可以将边索引的维度设置为 61,并且将其转换为 torch.tensor 类型。你可以尝试以下代码:

edge_index = torch.tensor([[0, 1, 0, 4, 0, 8, 1, 2, 1, 5, 2, 3, 2, 6, 3, 7, 3, 14, 4, 5, 4, 9, 5, 6, 5, 10, 6, 7, 6, 11, 7, 13, 8, 9, 8, 15, 9, 10, 9, 17, 10, 11, 10, 18, 11, 12, 12, 13, 12, 19, 13, 14, 13, 20, 14, 21, 15, 16, 16, 17, 27, 31, 28, 32, 29, 30, 29, 33, 30, 36, 31, 32, 31, 34, 32, 33, 32, 35, 33, 36, 34, 35, 35, 36]], dtype=torch.long).to(device)

然后再次运行你的代码,看看是否能够解决问题。

解决 PyTorch GCN 网络边索引超出范围的错误

原文地址: https://www.cveoy.top/t/topic/pgHB 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录