GNN 速查表

  • SparseTensor: 如果已勾选(✓),支持基于 torch_sparse.SparseTensor 的消息传递,例如GCNConv(...).forward(x, adj_t)。请参阅 这里 获取相关教程。

  • edge_weight: 如果勾选(✓),支持使用一维边权重信息进行消息传递,例如GraphConv(...).forward(x, edge_index, edge_weight)

  • edge_attr: 如果勾选(✓),支持使用多维边特征信息进行消息传递,例如GINEConv(...).forward(x, edge_index, edge_attr)

  • bipartite: 如果勾选(✓),支持在二分图中进行消息传递,源节点和目标节点可能具有不同的特征维度,例如SAGEConv(in_channels=(16, 32), out_channels=64)

  • static: 如果勾选(✓),支持在静态图中传递消息,例如GCNConv(...).forward(x, edge_index) 其中 x 的形状为 [batch_size, num_nodes, in_channels]

  • lazy: 如果勾选(✓),支持消息传递层的延迟初始化,例如SAGEConv(in_channels=-1, out_channels=64)

图神经网络操作符

名称

SparseTensor

edge_weight

edge_attr

二分图

静态

懒惰

SimpleConv

GCNConv (Paper)

ChebConv (Paper)

SAGEConv (Paper)

CuGraphSAGEConv (论文)

GraphConv (Paper)

GatedGraphConv (论文)

ResGatedGraphConv (Paper)

GATConv (Paper)

CuGraphGATConv (Paper)

FusedGATConv (论文)

GATv2Conv (论文)

TransformerConv (Paper)

AGNNConv (Paper)

TAGConv (Paper)

GINConv (Paper)

GINEConv (Paper)

ARMAConv (Paper)

SGConv (Paper)

SSGConv (论文)

APPNP (论文)

MFConv (Paper)

DNAConv (Paper)

GMMConv (Paper)

SplineConv (Paper)

NNConv (论文)

CGConv (论文)

EdgeConv (Paper)

FeaStConv (论文)

LEConv (Paper)

PNAConv (论文)

ClusterGCNConv (Paper)

GENConv (Paper)

GCN2Conv (Paper)

PANConv (Paper)

WLConv (论文)

WLConvContinuous (论文)

SuperGATConv (论文)

FAConv (Paper)

EGConv (Paper)

PDNConv (Paper)

GeneralConv (Paper)

LGConv (Paper)

GPSConv (Paper)

AntiSymmetricConv (论文)

DirGNNConv (论文)

MixHopConv (Paper)

异构图神经网络操作符

名称

SparseTensor

edge_weight

edge_attr

二分图

静态

懒惰

RGCNConv (Paper)

FastRGCNConv

CuGraphRGCNConv (论文)

RGATConv (论文)

FiLMConv (论文)

HGTConv (论文)

HEATConv (Paper)

HeteroConv

HANConv (论文)

超图神经网络操作符

名称

SparseTensor

edge_weight

edge_attr

二分图

静态

懒惰

HypergraphConv (论文)

点云神经网络操作符

名称

二分图

懒惰

GravNetConv (论文)

SignedConv (Paper)

PointNetConv (论文)

DynamicEdgeConv (Paper)

XConv (论文)

PPFConv (论文)

PointTransformerConv (Paper)

HeteroConv

PointGNNConv (论文)