mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
75 lines
1.7 KiB
ReStructuredText
75 lines
1.7 KiB
ReStructuredText
.. _apinn-mxnet:
|
|
|
|
dgl.nn (MXNet)
|
|
================
|
|
|
|
Conv Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.mxnet.conv.GraphConv
|
|
~dgl.nn.mxnet.conv.RelGraphConv
|
|
~dgl.nn.mxnet.conv.TAGConv
|
|
~dgl.nn.mxnet.conv.GATConv
|
|
~dgl.nn.mxnet.conv.EdgeConv
|
|
~dgl.nn.mxnet.conv.SAGEConv
|
|
~dgl.nn.mxnet.conv.SGConv
|
|
~dgl.nn.mxnet.conv.APPNPConv
|
|
~dgl.nn.mxnet.conv.GINConv
|
|
~dgl.nn.mxnet.conv.GatedGraphConv
|
|
~dgl.nn.mxnet.conv.GMMConv
|
|
~dgl.nn.mxnet.conv.ChebConv
|
|
~dgl.nn.mxnet.conv.AGNNConv
|
|
~dgl.nn.mxnet.conv.NNConv
|
|
|
|
Dense Conv Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.mxnet.conv.DenseGraphConv
|
|
~dgl.nn.mxnet.conv.DenseSAGEConv
|
|
~dgl.nn.mxnet.conv.DenseChebConv
|
|
|
|
Global Pooling Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.mxnet.glob.SumPooling
|
|
~dgl.nn.mxnet.glob.AvgPooling
|
|
~dgl.nn.mxnet.glob.MaxPooling
|
|
~dgl.nn.mxnet.glob.SortPooling
|
|
~dgl.nn.mxnet.glob.GlobalAttentionPooling
|
|
~dgl.nn.mxnet.glob.Set2Set
|
|
|
|
Heterogeneous Learning Modules
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.mxnet.HeteroGraphConv
|
|
|
|
Utility Modules
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.mxnet.utils.Sequential
|