在PyG中,单个graph定义为torch_geometric.data.Data实例,默认有以下属性:

    • data.x:节点特征矩阵,shape为[num_nodes, num_node_features]。
    • data.edge_index:COO格式的graph connectivity矩阵,shape为[2, num_edges],类型为torch.long。
    • data.edge_attr:边的特征矩阵,shape为[num_edges, num_edge_features]。
    • data.y:训练的target,shape不固定,比如,对于node-level任务,形状为[num_nodes, ],对于graph-level任务,形状为[1, ]。
    • data.pos:节点的位置(position)矩阵,shape为[num_nodes, num_dimensions]。