code: def __init__(self, G, in_size, hidden_size, out_size):
      super(HeteroRGCN, self).__init__()
      # create layers
      self.layer1 = HeteroRGCNLayer(in_size, hidden_size, G.etypes)
      self.layer2 = HeteroRGCNLayer(hidden_size, out_size, G.etypes)
  
imports: import dgl
  import math
  import torch
  import torch.nn
  import torch.nn.functional
  import dgl.function
  from dgl.nn.functional import edge_softmax
label_desc: Use image node_id 1 to create a new HeteroRGCN object from inherited base classes: nn with example: obj = HeteroRGCN(G, in_size, hidden_size, out_size)