def METHOD_NAME(self, node):
    alpha = torch.softmax(node.mailbox['a'], dim=1)
    alpha = F.dropout(alpha, p=self.dropout, training=self.training)
    x = (node.mailbox['m'] * alpha.unsqueeze(-1)).sum(dim=1)
    return {'x': x}