dgl.nn (TensorFlow)

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

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

GlobalAttentionPooling

Global Attention Pooling from Gated Graph Sequence Neural Networks

Heterogeneous Learning Modules