mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-07 20:41:20 +08:00
30 lines
584 B
ReStructuredText
30 lines
584 B
ReStructuredText
.. _apinn:
|
|
|
|
dgl.nn
|
|
==========
|
|
|
|
.. automodule:: dgl.nn
|
|
|
|
.. toctree::
|
|
|
|
nn.pytorch
|
|
nn.mxnet
|
|
nn.tensorflow
|
|
|
|
dgl.nn.functional
|
|
=================
|
|
|
|
Edge Softmax module
|
|
-------------------
|
|
|
|
We also provide framework agnostic edge softmax module which was frequently used in
|
|
GNN-like structures, e.g.
|
|
`Graph Attention Network <https://arxiv.org/pdf/1710.10903.pdf>`_,
|
|
`Transformer <https://papers.nips.cc/paper/7181-attention-is-all-you-need.pdf>`_,
|
|
`Capsule <https://arxiv.org/pdf/1710.09829.pdf>`_, etc.
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
functional.edge_softmax
|