Modified LightGCN Convolutional Layer

LightGCNConv.py

class LightGCNConv(MessagePassing):
    def __init__(self, aggregation='mean', **kwargs):  
        super(LightGCNConv, self).__init__(**kwargs)
        self.aggregation = aggregation

    def forward(self, x, edge_index, size = None):
        out = self.propagate(edge_index, x=(x, x))
        return out

    def message(self, x_j):
        out = x_j
        return out

    def aggregate(self, inputs, index, dim_size = None):
        node_dim = self.node_dim
        out = torch_scatter.scatter(inputs, index, dim=node_dim, reduce=self.aggregation)
        return out