nn.mxnet.rst 615 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
.. _apinn-mxnet:

dgl.nn.mxnet
============

dgl.nn.mxnet.conv
-----------------

.. automodule:: dgl.nn.mxnet.conv

.. autoclass:: dgl.nn.mxnet.conv.GraphConv
    :members: weight, bias, forward
    :show-inheritance:
14

Minjie Wang's avatar
Minjie Wang committed
15
16
17
18
.. autoclass:: dgl.nn.mxnet.conv.RelGraphConv
    :members: forward
    :show-inheritance:

19
20
21
22
.. autoclass:: dgl.nn.mxnet.conv.TAGConv
    :members: forward
    :show-inheritance:

23
24
25
26
27
dgl.nn.mxnet.glob
-----------------

.. automodule:: dgl.nn.mxnet.glob
    :members:
Zihao Ye's avatar
Zihao Ye committed
28
    :show-inheritance:
29
30
31
32
33

dgl.nn.mxnet.softmax
--------------------

.. automodule:: dgl.nn.mxnet.softmax
Zihao Ye's avatar
Zihao Ye committed
34
    :members: edge_softmax