Pytorch Geometric

图的数据处理

图用于建模对象（节点）之间的成对关系（边）。Pytorch Geometric 中的单个图形由<code style="color:red;">torch_geometric.data.Data</code> 的实例描述，该实例默认情况下具有以下属性：

• <code style="color:red;">data.x</code>: <code style="color:red;">[num_nodes, num_node_features</code>，节点特征矩阵
• <code style="color:red;">data.edge_index</code>: <code style="color:red;">[2, num_edges]</code>，表示节点间的连接关系（边），<code style="color:red;">edge_index[0]</code> 是源节点，<code style="color:red;">edge_index[1]</code> 是目标节点
• <code style="color:red;">data.edge_attr</code>: <code style="color:red;">[num_edges, num_edge_features]</code>，边的特征矩阵
• <code style="color:red;">data.y</code>: 节点级别的<code style="color:red;">label，[num_nodes, ]</code>，或图级别的<code style="color:red;">label，[1, ]</code>
• <code style="color:red;">data.pos</code>: <code style="color:red;">[num_nodes, num_dimensions]</code>，节点位置矩阵

这些属性都不是必需的。实际上，<code style="color:red;">Data</code>对象甚至不限于这些属性。例如，我们可以通过data.face对其进行扩展，以将3D网格中的三角形的连通性保存在形状为<code style="color:red;">[3，num_faces]</code>的张量中，并键入<code style="color:red;">torch.long</code>。