class_docstr: Generates model graph, each node is created from single or multiple jit trace nodes.
docstr: Extract the auxiliary information for the nodegroups
  if necessary. For example, view/flatten operations may
  need the shape of the input tensor and output tensor.
code: def _extract_auxiliary_info(self):
      """
      Extract the auxiliary information for the nodegroups
      if necessary. For example, view/flatten operations may
      need the shape of the input tensor and output tensor.
      """
      # extract the input & output shape for the view and flatten
      for node_group in self.nodes_py.nodes_op:
          if node_group.op_type in [
              "aten::view",
              "aten::flatten",
              "aten::mean",
              "aten::reshape",
          ]:
              # get shape infor for view (aten::view) func
              cpp_node = list(
                  filter(
                      lambda x: x.kind() == node_group.op_type,
                      node_group.node_cpps,
                  )
              )[0]
              node_group.auxiliary = self._extract_shape_info(cpp_node)
          elif node_group.op_type == "Linear":
              node_group.auxiliary = self._extract_linear_shape_info(
                  node_group
              )
          elif node_group.op_type == CAT_KIND:
              # get the detail information for cat func
              cpp_node = list(
                  filter(
                      lambda x: x.kind() == node_group.op_type,
                      node_group.node_cpps,
                  )
              )[0]
              node_group.auxiliary = self._extract_cat_info(
                  node_group, cpp_node
              )
  
imports: import logging
  import queue
  import re
  from collections import defaultdict
  import torch
  from torch.utils.tensorboard._pytorch_graph import NodePy, NodePyIO, NodePyOP, GraphPy
label_desc: Use image node_id 13 for calling the TorchModuleGraph obj's underlying member method code with example usage: obj._extract_auxiliary_info() without return types