def METHOD_NAME(self, node_feat_dim, message_net_hiddens):
    layer = []
    layer.append(torch.nn.Linear(node_feat_dim * 2, message_net_hiddens[0]))
    for i in range(1, len(message_net_hiddens)):
        layer.append(torch.nn.ReLU())
        layer.append(torch.nn.Linear(message_net_hiddens[i - 1], message_net_hiddens[i]))
    self.message_net = torch.nn.Sequential(*layer)