code: def forward(self, g, lg, x, lg_x, deg_g, deg_lg, pm_pd):
      next_x = self.g_layer(g, x, lg_x, deg_g, pm_pd)
      pm_pd_y = th.transpose(pm_pd, 0, 1)
      next_lg_x = self.lg_layer(lg, lg_x, x, deg_lg, pm_pd_y)
      return next_x, next_lg_x
  
imports: import os
  import dgl
  import torch
  import torch
  import torch.nn
  import torch.nn.functional
  from dgl.data import citation_graph
  import matplotlib.pyplot
  import networkx
  import dgl.function
  from torch.utils.data import DataLoader
label_desc: Use image node_id 2 for calling the LGNNLayer obj's underlying member method code with example usage: obj.forward(g, lg, x, lg_x, deg_g, deg_lg, pm_pd) and returns: next_x,  next_lg_x