【冰糖Python】PyTorch Geometric：Data 对象

PyTorch Geometric 中提供了一种使用多种属性表示图的数据类型 Data

class Data(x=None, edge_index=None, edge_attr=None, y=None, pos=None, normal=None, face=None, **kwargs)

x：tensor 类型，表示形状为 [num_nodes, num_node_features] 的结点的特征矩阵，默认为 None（num_nodes 为结点数，详见类属性部分）

edge_index：long tensor 类型，表示形状为 [2, num_edges] 的图的连接性矩阵，即表示边的矩阵，默认为 None（对于无向边，通过定义两个方向的边实现）

edge_attr：tensor 类型，表示形状为 [num_edges, num_edge_features] 的边的特征矩阵，默认为 None

y：tensor 类型，任意形状的图或结点的标签，默认为 None

pos：tensor 类型，表示形状为 [num_nodes, num_dimensions] 的结点的位置矩阵

normal：tensor 类型，表示形状为 [num_nodes, num_dimensions] 的法向量矩阵

face：long tensor 类型，表示形状为 [3, num_faces