Skip to content

Latest commit

 

History

History
169 lines (142 loc) · 4.77 KB

nn-pytorch.rst

File metadata and controls

169 lines (142 loc) · 4.77 KB
Raw

dgl.nn (PyTorch)

Conv Layers

~dgl.nn.pytorch.conv.GraphConv ~dgl.nn.pytorch.conv.EdgeWeightNorm ~dgl.nn.pytorch.conv.RelGraphConv ~dgl.nn.pytorch.conv.TAGConv ~dgl.nn.pytorch.conv.GATConv ~dgl.nn.pytorch.conv.GATv2Conv ~dgl.nn.pytorch.conv.EGATConv ~dgl.nn.pytorch.conv.EdgeGATConv ~dgl.nn.pytorch.conv.EdgeConv ~dgl.nn.pytorch.conv.SAGEConv ~dgl.nn.pytorch.conv.SGConv ~dgl.nn.pytorch.conv.APPNPConv ~dgl.nn.pytorch.conv.GINConv ~dgl.nn.pytorch.conv.GINEConv ~dgl.nn.pytorch.conv.GatedGraphConv ~dgl.nn.pytorch.conv.GatedGCNConv ~dgl.nn.pytorch.conv.GMMConv ~dgl.nn.pytorch.conv.ChebConv ~dgl.nn.pytorch.conv.AGNNConv ~dgl.nn.pytorch.conv.NNConv ~dgl.nn.pytorch.conv.AtomicConv ~dgl.nn.pytorch.conv.CFConv ~dgl.nn.pytorch.conv.DotGatConv ~dgl.nn.pytorch.conv.TWIRLSConv ~dgl.nn.pytorch.conv.TWIRLSUnfoldingAndAttention ~dgl.nn.pytorch.conv.GCN2Conv ~dgl.nn.pytorch.conv.HGTConv ~dgl.nn.pytorch.conv.GroupRevRes ~dgl.nn.pytorch.conv.EGNNConv ~dgl.nn.pytorch.conv.PNAConv ~dgl.nn.pytorch.conv.DGNConv

CuGraph Conv Layers

~dgl.nn.pytorch.conv.CuGraphRelGraphConv ~dgl.nn.pytorch.conv.CuGraphGATConv ~dgl.nn.pytorch.conv.CuGraphSAGEConv

Dense Conv Layers

~dgl.nn.pytorch.conv.DenseGraphConv ~dgl.nn.pytorch.conv.DenseSAGEConv ~dgl.nn.pytorch.conv.DenseChebConv

Global Pooling Layers

~dgl.nn.pytorch.glob.SumPooling ~dgl.nn.pytorch.glob.AvgPooling ~dgl.nn.pytorch.glob.MaxPooling ~dgl.nn.pytorch.glob.SortPooling ~dgl.nn.pytorch.glob.WeightAndSum ~dgl.nn.pytorch.glob.GlobalAttentionPooling ~dgl.nn.pytorch.glob.Set2Set ~dgl.nn.pytorch.glob.SetTransformerEncoder ~dgl.nn.pytorch.glob.SetTransformerDecoder

Score Modules for Link Prediction and Knowledge Graph Completion ----------------------------------------

~dgl.nn.pytorch.link.EdgePredictor ~dgl.nn.pytorch.link.TransE ~dgl.nn.pytorch.link.TransR

Heterogeneous Learning Modules

~dgl.nn.pytorch.HeteroGraphConv ~dgl.nn.pytorch.HeteroLinear ~dgl.nn.pytorch.HeteroEmbedding ~dgl.nn.pytorch.TypedLinear

Utility Modules

~dgl.nn.pytorch.utils.Sequential ~dgl.nn.pytorch.utils.WeightBasis ~dgl.nn.pytorch.factory.KNNGraph ~dgl.nn.pytorch.factory.SegmentedKNNGraph ~dgl.nn.pytorch.factory.RadiusGraph ~dgl.nn.pytorch.utils.JumpingKnowledge ~dgl.nn.pytorch.sparse_emb.NodeEmbedding ~dgl.nn.pytorch.explain.GNNExplainer ~dgl.nn.pytorch.explain.HeteroGNNExplainer ~dgl.nn.pytorch.explain.SubgraphX ~dgl.nn.pytorch.explain.HeteroSubgraphX ~dgl.nn.pytorch.explain.PGExplainer ~dgl.nn.pytorch.explain.HeteroPGExplainer ~dgl.nn.pytorch.utils.LabelPropagation ~dgl.nn.pytorch.graph_transformer.DegreeEncoder ~dgl.nn.pytorch.utils.LaplacianPosEnc ~dgl.nn.pytorch.graph_transformer.BiasedMultiheadAttention ~dgl.nn.pytorch.graph_transformer.EGTLayer ~dgl.nn.pytorch.graph_transformer.GraphormerLayer ~dgl.nn.pytorch.graph_transformer.PathEncoder ~dgl.nn.pytorch.graph_transformer.SpatialEncoder ~dgl.nn.pytorch.graph_transformer.SpatialEncoder3d

Network Embedding Modules

~dgl.nn.pytorch.DeepWalk ~dgl.nn.pytorch.MetaPath2Vec

Utility Modules for Graph Transformer

~dgl.nn.pytorch.gt.DegreeEncoder ~dgl.nn.pytorch.gt.LapPosEncoder ~dgl.nn.pytorch.gt.PathEncoder ~dgl.nn.pytorch.gt.SpatialEncoder ~dgl.nn.pytorch.gt.SpatialEncoder3d ~dgl.nn.pytorch.gt.BiasedMHA ~dgl.nn.pytorch.gt.GraphormerLayer ~dgl.nn.pytorch.gt.EGTLayer