nn-mxnet.rst

dgl.nn (MXNet)

Conv Layers

~dgl.nn.mxnet.conv.GraphConv ~dgl.nn.mxnet.conv.RelGraphConv ~dgl.nn.mxnet.conv.TAGConv ~dgl.nn.mxnet.conv.GATConv ~dgl.nn.mxnet.conv.EdgeConv ~dgl.nn.mxnet.conv.SAGEConv ~dgl.nn.mxnet.conv.SGConv ~dgl.nn.mxnet.conv.APPNPConv ~dgl.nn.mxnet.conv.GINConv ~dgl.nn.mxnet.conv.GatedGraphConv ~dgl.nn.mxnet.conv.GMMConv ~dgl.nn.mxnet.conv.ChebConv ~dgl.nn.mxnet.conv.AGNNConv ~dgl.nn.mxnet.conv.NNConv

Dense Conv Layers

~dgl.nn.mxnet.conv.DenseGraphConv ~dgl.nn.mxnet.conv.DenseSAGEConv ~dgl.nn.mxnet.conv.DenseChebConv

Global Pooling Layers

~dgl.nn.mxnet.glob.SumPooling ~dgl.nn.mxnet.glob.AvgPooling ~dgl.nn.mxnet.glob.MaxPooling ~dgl.nn.mxnet.glob.SortPooling ~dgl.nn.mxnet.glob.GlobalAttentionPooling ~dgl.nn.mxnet.glob.Set2Set

Heterogeneous Learning Modules

~dgl.nn.mxnet.HeteroGraphConv

Utility Modules

~dgl.nn.mxnet.utils.Sequential