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)。
图神经网络操作符
名称 |
|
|
|
二分图 |
静态 |
懒惰 |
|---|---|---|---|---|---|---|
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
||||||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
||||||
✓ |
||||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
✓ |
||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
✓ |
||||
✓ |
✓ |
|||||
✓ |
✓ |
|||||
|
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
异构图神经网络操作符
名称 |
|
|
|
二分图 |
静态 |
懒惰 |
|---|---|---|---|---|---|---|
✓ |
||||||
✓ |
||||||
✓ |
||||||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
✓ |
|||
✓ |
✓ |
|||||
✓ |
✓ |
✓ |
||||
✓ |
||||||
✓ |
✓ |
超图神经网络操作符
名称 |
|
|
|
二分图 |
静态 |
懒惰 |
|---|---|---|---|---|---|---|
✓ |
✓ |
✓ |
点云神经网络操作符
名称 |
二分图 |
懒惰 |
|---|---|---|
✓ |
✓ |
|
✓ |
✓ |
|
✓ |
||
✓ |
||
✓ |
||
✓ |
✓ |
|