code: def _get_message_net(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)
  
file_path: DIG/dig/auggraph/method/GraphAug/aug/aug_encoder.py
imports: import torch
  from torch.nn import Linear, Sequential, ReLU, BatchNorm1d
  from torch_geometric.nn import GINConv
  from torch_scatter import scatter
  from ..constants import *
label_desc: Use image node_id 2 for calling the GEMBConv obj's underlying member method code with example usage: obj._get_message_net(node_feat_dim, message_net_hiddens) without return types