dgl.nn (MXNet)¶
Conv Layers¶
Dense Conv Layers¶
Graph Convolutional layer from Semi-Supervised Classification with Graph Convolutional Networks |
|
GraphSAGE layer from Inductive Representation Learning on Large Graphs |
|
Chebyshev Spectral Graph Convolution layer from Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering |
Global Pooling Layers¶
Apply sum pooling over the nodes in the graph. |
|
Apply average pooling over the nodes in the graph. |
|
Apply max pooling over the nodes in the graph. |
|
Pooling layer from An End-to-End Deep Learning Architecture for Graph Classification |
|
Global Attention Pooling layer from Gated Graph Sequence Neural Networks |
|
Set2Set operator from Order Matters: Sequence to sequence for sets |
Heterogeneous Learning Modules¶
A generic module for computing convolution on heterogeneous graphs |
Utility Modules¶
A squential container for stacking graph neural network blocks |