dmlc--dgl
Tingyu Wang
bfd411d0d5
* add CuGraphGATConv model * lintrunner * update model to reflect changes in make_mfg_csr(), move max_in_degree to forward() * simplify pytest markers * fall back to FG option for large fanout * update error msg * add feat_drop and activation options * add residual option * Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * Update python/dgl/nn/pytorch/conv/cugraph_gatconv.py Co-authored-by: Mufei Li <mufeili1996@gmail.com> * reset res_fc --------- Co-authored-by: Mufei Li <mufeili1996@gmail.com>
148 行
4.1 KiB
ReStructuredText
148 行
4.1 KiB
ReStructuredText
.. _apinn-pytorch:
|
|
|
|
dgl.nn (PyTorch)
|
|
================
|
|
|
|
Conv Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~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.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.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
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.pytorch.conv.CuGraphRelGraphConv
|
|
~dgl.nn.pytorch.conv.CuGraphGATConv
|
|
~dgl.nn.pytorch.conv.CuGraphSAGEConv
|
|
|
|
Dense Conv Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.pytorch.conv.DenseGraphConv
|
|
~dgl.nn.pytorch.conv.DenseSAGEConv
|
|
~dgl.nn.pytorch.conv.DenseChebConv
|
|
|
|
Global Pooling Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~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
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.pytorch.link.EdgePredictor
|
|
~dgl.nn.pytorch.link.TransE
|
|
~dgl.nn.pytorch.link.TransR
|
|
|
|
Heterogeneous Learning Modules
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.pytorch.HeteroGraphConv
|
|
~dgl.nn.pytorch.HeteroLinear
|
|
~dgl.nn.pytorch.HeteroEmbedding
|
|
~dgl.nn.pytorch.TypedLinear
|
|
|
|
Utility Modules
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~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.utils.LabelPropagation
|
|
~dgl.nn.pytorch.graph_transformer.DegreeEncoder
|
|
~dgl.nn.pytorch.utils.LaplacianPosEnc
|
|
~dgl.nn.pytorch.graph_transformer.BiasedMultiheadAttention
|
|
~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
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.pytorch.DeepWalk
|
|
~dgl.nn.pytorch.MetaPath2Vec
|