code: def get_target_input(module, input, output):
      
      try:
          del module.target_input
      except AttributeError:
          pass
      setattr(module, 'target_input', input)
docs: A forward hook which saves the tensor - attached to its graph.
      Used if we want to explain the interim outputs of a model