自定义PyTorch Geometric数据集时遭遇edge_index索引错误求助

这个报错的核心是你的edge_index中存在索引超出节点特征矩阵x的有效范围，即便你觉得已经确认过最大值不超过节点数，也可能是以下细节没处理到位：

• 节点数的判断错误：你统计的节点数和x的实际第一维度长度不一致。比如你以为有15个节点，但x.shape[0]实际是13，这时候哪怕edge_index的最大值是14，也会触发这个报错。解决时直接打印x.shape[0]和edge_index.max()对比，不要凭主观统计的节点数判断。
• 索引没有转成0-based：如果你的原始数据是1-based编号（比如节点从1开始计数），转成PyTorch Geometric要求的0-based时没处理彻底。比如节点数是13，1-based的最大节点号是13，转成0-based应该是12，若edge_index里还保留13，就会超出[0,13)的范围。检查原始数据的索引规则，统一转成0-based后再构建edge_index。
• 自定义数据集构建Data对象时的逻辑错误：比如不小心截断了x（比如处理特征时过滤了部分节点，但没同步更新edge_index），或者x的维度顺序搞反了（比如把x做成了(特征数, 节点数)而不是(节点数, 特征数)），这会让PyTorch Geometric误判节点数量。打印Data对象的完整信息（print(data)），核对x.shape、edge_index.shape和edge_index.max()是否匹配。
• 批次加载时的自定义逻辑问题：如果你自己实现了collate_fn，可能在合并多个Data对象时出错，导致合并后的x和edge_index不对应。对比InMemory数据集的默认collate逻辑，检查自己的代码是否正确处理了节点索引的偏移（PyTorch Geometric默认会自动偏移不同图的节点索引）。

内容的提问来源于stack exchange，提问作者Aster Bot