dgl.nn (MXNet)

Global Pooling Layers

SumPooling

Apply sum pooling over the nodes in the graph.

AvgPooling

Apply average pooling over the nodes in the graph.

MaxPooling

Apply max pooling over the nodes in the graph.

SortPooling

Pooling layer from An End-to-End Deep Learning Architecture for Graph Classification

GlobalAttentionPooling

Global Attention Pooling layer from Gated Graph Sequence Neural Networks

Set2Set

Set2Set operator from Order Matters: Sequence to sequence for sets

Heterogeneous Learning Modules

HeteroGraphConv

A generic module for computing convolution on heterogeneous graphs

Utility Modules

Sequential

A squential container for stacking graph neural network blocks