gpool¶
Simple Pooling Layer. Allows you to reduce a 3 dimensional Tensor (-1,gs,param) into (-1,param) by running a Pooling Operation on each Node. Is the simplest way to finish a classical Graph Network that does not break Graph Permutation Symmetry
Arguments
- gs: The Number of Nodes of the Graph (Graph Size)
- param: The Number of Features for each Node
- mode=”max”: Pooling Operation, either “max”,”mean” or “sum”