mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-07 20:41:20 +08:00
131 lines
2.8 KiB
ReStructuredText
131 lines
2.8 KiB
ReStructuredText
.. _apinn-pytorch:
|
|
|
|
dgl.nn.pytorch
|
|
==============
|
|
|
|
dgl.nn.pytorch.conv
|
|
-------------------
|
|
|
|
.. automodule:: dgl.nn.pytorch.conv
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GraphConv
|
|
:members: weight, bias, forward, reset_parameters
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.RelGraphConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.TAGConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GATConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.EdgeConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.SAGEConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.SGConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.APPNPConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GINConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GatedGraphConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.GMMConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.ChebConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.AGNNConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.NNConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.DenseGraphConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.DenseSAGEConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.conv.DenseChebConv
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
dgl.nn.pytorch.glob
|
|
-------------------
|
|
.. automodule:: dgl.nn.pytorch.glob
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.SumPooling
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.AvgPooling
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.MaxPooling
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.SortPooling
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.GlobalAttentionPooling
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.Set2Set
|
|
:members: forward
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.SetTransformerEncoder
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. autoclass:: dgl.nn.pytorch.glob.SetTransformerDecoder
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
dgl.nn.pytorch.softmax
|
|
----------------------
|
|
|
|
.. automodule:: dgl.nn.pytorch.softmax
|
|
:members: edge_softmax
|
|
|
|
dgl.nn.pytorch.factory
|
|
----------------------
|
|
|
|
.. automodule:: dgl.nn.pytorch.KNNGraph
|
|
:members:
|
|
:show-inheritance:
|
|
|
|
.. automodule:: dgl.nn.pytorch.SegmentedKNNGraph
|
|
:members:
|
|
:show-inheritance:
|