torch_geometric.transforms.VirtualNode
- class VirtualNode[source]
Bases:
BaseTransform将一个虚拟节点附加到给定的同质图中,该节点连接到所有其他节点,如“Neural Message Passing for Quantum Chemistry”论文中所述(功能名称:
virtual_node)。虚拟节点充当全局暂存空间,每个节点在消息传递的每一步都从中读取和写入。这使得信息在传播阶段能够长距离传递。虚拟节点的节点和边特征被添加为零填充的输入特征。 此外,将为虚拟节点的传入和传出信息添加特殊的边类型。