code: def _get_message_net(self, node_feat_dim, message_net_hiddens):
      layer = []
      layer.append(nn.Linear(node_feat_dim * 2, message_net_hiddens[0]))
      for i in range(1, len(message_net_hiddens)):
          layer.append(nn.ReLU())
          layer.append(
              nn.Linear(
                  message_net_hiddens[i - 1], message_net_hiddens[i]
              )
          )
      self.message_net = nn.Sequential(*layer)
  
imports: import torch
  import torch.nn
  import torch.nn.functional
  from torch_scatter import scatter
  from .genet import Readout
  from dig.auggraph.method.GraphAug.constants import *
label_desc: Use image node_id 2 for calling the GMNConv obj's underlying member method code with example usage: obj._get_message_net(node_feat_dim, message_net_hiddens) without return types