code: def __init__(self, in_size, out_size, etypes):
      super(HeteroRGCNLayer, self).__init__()
      # W_r for each relation
      self.weight = nn.ModuleDict(
          {name: nn.Linear(in_size, out_size) for name in 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 HeteroRGCNLayer object from inherited base classes: nn with example: obj = HeteroRGCNLayer(in_size, out_size, etypes)