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.