code: def _add_forward_tensor(self, name: str, inp: any, outp: any):
      self._forward_input[name] = inp
      self._forward_output[name] = outp
  
imports: from typing import Callable, Dict, List, TYPE_CHECKING
  from labml.internal.util.strings import is_pattern_match
label_desc: Use image node_id 2 for calling the ModelProbe obj's underlying member method code with example usage: obj._add_forward_tensor(name, inp, outp) without return types