class_docstr: Feature Hook Helper
  This module helps with the setup and extraction of hooks for extracting features from
  internal nodes in a model by node name. This works quite well in eager Python but needs
  redesign for torcscript.
class_docstr_tok: ["Feature","Hook","Helper","This","module","helps","with","the","setup","and","extraction","of","hooks","for","extracting","features","from","internal","nodes","in","a","model","by","node","name",".","This","works","quite","well","in","eager","Python","but","needs","redesign","for","torcscript","."]
code: def get_output(self, device) -> Dict[str, torch.tensor]:
      output = self._feature_outputs[device]
      self._feature_outputs[
          device
      ] = OrderedDict()  # clear after reading
      return output
  
imports: from collections import OrderedDict, defaultdict
  from copy import deepcopy
  from functools import partial
  from typing import Dict, List, Tuple
  import torch
  import torch.nn
label_desc: Use image node_id 3 for calling the FeatureHooks obj's underlying member method code with example usage: obj.get_output(device) and returns: output