Method __init__

tf_euler/python/graph_pool/base_pool.py:22–24

(self, aggr='add')

class Pooling(object):
    def __init__(self, aggr='add'):
        assert aggr in ['add', 'mean', 'max']
        self.aggr = aggr

    def __call__(self, inputs, index, size=None):
        size = tf.reduce_max(index) + 1 if size is None else size