SortPoolingΒΆ

class dgl.nn.tensorflow.glob.SortPooling(*args, **kwargs)[source]ΒΆ

Bases: tensorflow.python.keras.engine.base_layer.Layer

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

Parameters

k (int) – The number of nodes to hold for each graph.