code: def __init__(self, in_nodes, out_nodes, f_size):
      super(DGLRoutingLayer, self).__init__()
      self.g = init_graph(in_nodes, out_nodes, f_size)
      self.in_nodes = in_nodes
      self.out_nodes = out_nodes
      self.in_indx = list(range(in_nodes))
      self.out_indx = list(range(in_nodes, in_nodes + out_nodes))
  
file_path: dgl/tutorials/models/4_old_wines/2_capsule.py
imports: import os
  import dgl
  import matplotlib.pyplot
  import numpy
  import torch
  import torch.nn
  import torch.nn.functional
  import dgl.function
  import matplotlib.animation
  import seaborn
  import networkx
  from networkx.algorithms import bipartite
label_desc: Use image node_id 1 to create a new DGLRoutingLayer object from inherited base classes: nn with example: obj = DGLRoutingLayer(in_nodes, out_nodes, f_size)