PyTorch Geometric中Data(edge_index=[2,4],x=[3,1])代码含义咨询

这行内容不是可执行代码，而是PyTorch Geometric里Data对象的字符串打印输出——当你运行print(data)时，就会得到这样的结果，用来快速展示图数据的核心张量形状：

• edge_index=[2, 4]：表示edge_index张量的形状是(2, 4)：
  • 第一个维度2对应边的「源节点」和「目标节点」两个维度（PyTorch Geometric要求edge_index必须是[2, num_edges]的格式）；
  • 第二个维度4表示图中总共有4条边（对应代码里定义的4条无向边：0↔1、1↔2各两条）。
• x=[3, 1]：表示节点特征张量x的形状是(3, 1)：
  • 第一个维度3是图中的节点总数（对应图里的3个节点）；
  • 第二个维度1是每个节点的特征维度（代码里每个节点用1维数值作为特征）。

简单来说，这行输出是帮你快速确认当前图的核心结构：3个节点（每个带1维特征）、4条边，无需手动逐个打印张量的.shape属性。

内容的提问来源于stack exchange，提问作者Krishn Nand