Method apply_node

tf_euler/python/convolution/relation_conv.py:72–73

(self, x, aggr_out)

        return tf.reshape(res, [-1, self.dim])

    def apply_node(self, x, aggr_out):
        return self.fc(x) + aggr_out