dmlc--dgl
JitongZ
731cd90ad3
* Fix typo in nn-tensorflow.rst Fixed typo in the link to the HeteroGraphConv module * Update nn-tensorflow.rst Fixed typo in the link to HeteroGraphConv module
46 行
1.1 KiB
ReStructuredText
46 行
1.1 KiB
ReStructuredText
.. _apinn-tensorflow:
|
|
|
|
dgl.nn (TensorFlow)
|
|
================
|
|
|
|
Conv Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.tensorflow.conv.GraphConv
|
|
~dgl.nn.tensorflow.conv.RelGraphConv
|
|
~dgl.nn.tensorflow.conv.GATConv
|
|
~dgl.nn.tensorflow.conv.SAGEConv
|
|
~dgl.nn.tensorflow.conv.ChebConv
|
|
~dgl.nn.tensorflow.conv.SGConv
|
|
~dgl.nn.tensorflow.conv.APPNPConv
|
|
~dgl.nn.tensorflow.conv.GINConv
|
|
|
|
Global Pooling Layers
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.tensorflow.glob.SumPooling
|
|
~dgl.nn.tensorflow.glob.AvgPooling
|
|
~dgl.nn.tensorflow.glob.MaxPooling
|
|
~dgl.nn.tensorflow.glob.SortPooling
|
|
~dgl.nn.tensorflow.glob.GlobalAttentionPooling
|
|
|
|
Heterogeneous Learning Modules
|
|
----------------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
~dgl.nn.tensorflow.HeteroGraphConv
|