code: def gen_Input(self):
      IR_node = self.IR_graph.node.add()
      IR_node.name = "input"
      IR_node.op = "DataInput"
  
      for node in self.IR_graph.node:
          if node.name in self.src_graph.input_layers:
              node.input.append("input")
  
      assert len(self.input_shape) == 4
      new_dim = IR_node.attr["shape"].shape.dim.add()
      if self.input_shape[0] == 1:
          new_dim.size = -1
      else:
          new_dim.size = self.input_shape[0]
      for index in [2, 3, 1]:
          new_dim = IR_node.attr["shape"].shape.dim.add()
          new_dim.size = self.input_shape[index]
  
      shape = graph_pb2.TensorShape()
      new_dim = shape.dim.add()
      shape_pytorch = self.input_shape
  
      if len(shape_pytorch) == 4:
          if shape_pytorch[0] == 1:
              new_dim.size = -1
          else:
              new_dim.size = shape_pytorch[0]
          for index in [2, 3, 1]:
              new_dim = shape.dim.add()
              dim = shape_pytorch[index]
              new_dim.size = dim if dim else -1
      elif len(shape_pytorch) == 2:
          if shape_pytorch[0] == 1:
              new_dim.size = -1
          else:
              new_dim.size = shape_pytorch[0]
          for _ in range(2):
              new_dim = shape.dim.add()
              new_dim.size = 1
          new_dim = shape.dim.add()
          dim = shape_pytorch[1]
          new_dim.size = dim if dim else -1
  
      IR_node.attr["_output_shapes"].list.shape.extend([shape])
  
imports: import os
  import numpy
  import mmdnn.conversion.common.IR.graph_pb2
  from mmdnn.conversion.common.IR.graph_pb2 import NodeDef, GraphDef, DataType
  from mmdnn.conversion.common.utils import *
  from mmdnn.conversion.common.DataStructure.parser import Parser
  from mmdnn.conversion.pytorch.pytorch_graph import PytorchGraph040
  from mmdnn.conversion.pytorch.pytorch_graph import PytorchGraph151
  import torch
  import torchvision
label_desc: Use image node_id 8 for calling the PytorchParser obj's underlying member method code with example usage: obj.gen_Input() without return types