PyTorch Geometric示例代码中Data类参数及输出含义问询

edge_index维度[2, 4]的含义 #

edge_index采用COO稀疏格式存储图的连接关系：

• 第一维的2是固定值，两行分别对应所有边的源节点索引序列和目标节点索引序列
• 第二维的4代表当前图总共定义了4条边，你说的「定义了4条从0开始编号的边」的假设完全正确，代码中对应的4条边为0→1、1→0、1→2、2→1，属于无向图中每条无向边存储双向指向的标准写法。

x维度[3, 1]的含义 #

x是节点特征矩阵，两个维度分别对应：

• 第一个值3：当前图的总节点数，这里共有编号为0、1、2的3个节点
• 第二个值1：每个节点的特征向量维度，你代码中每个节点仅定义了1个浮点型特征，因此特征维度为1。

内容的提问来源于stack exchange，提问作者Astra Uvarova - Saturn's star