Method __call__

tf_euler/python/graph_pool/base_pool.py:26–29

(self, inputs, index, size=None)

        self.aggr = aggr

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