code: def forward(self, g, in_feat):
      h = self.conv1(
          g, in_feat, torch.ones(g.num_edges(), 1).to(g.device)
      )
      h = F.relu(h)
      h = self.conv2(g, h, torch.ones(g.num_edges(), 1).to(g.device))
      return h
  
imports: import os
  import dgl
  import dgl.function
  import torch
  import torch.nn
  import torch.nn.functional
  import dgl.data
label_desc: Use image node_id 2 for calling the Model obj's underlying member method code with example usage: obj.forward(g, in_feat) and returns: h