mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-06 20:04:24 +08:00
6c23fba8fa
* move edge_softmax to functional * change examples * fixes * revert * fix * fix * remove * fix * test * add init Co-authored-by: Minjie Wang <wmjlyjemaine@gmail.com>
30 lines
587 B
ReStructuredText
30 lines
587 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/
|
|
|
|
nn.functional.edge_softmax
|