Method __call__

tf_euler/python/convolution/graph_conv.py:33–40

(self, x, edge_index, size=None, **kwargs)

        self.liner = tf.layers.Dense(dim, use_bias=True)

    def __call__(self, x, edge_index, size=None, **kwargs):
        h = [None if x[0] is None else self.fc(x[0]),
             None if x[1] is None else self.fc(x[1])]
        gather_x, gather_h = self.gather_feature([x, h], edge_index)
        out = self.apply_edge(gather_h[1])
        out = mp_ops.scatter_(self.aggr, out, edge_index[0], size=size[0])
        out = self.apply_node(out, x[0])
        return out

    def apply_edge(self, x_j):
        return x_j