def METHOD_NAME(self, data):
    x, pos, batch = data.x, data.pos, data.batch
    x0 = torch.cat([x, pos], dim=-1)
    x1 = self.conv1(x0, batch)
    x2 = self.conv2(x1, batch)
    x3 = self.conv3(x2, batch)
    out = self.mlp(torch.cat([x1, x2, x3], dim=1))
    return F.log_softmax(out, dim=1)