code: def to_class_name(cls, type_name) -> Optional[str]:
      if type_name.startswith("__torch__."):
          return type_name[len("__torch__.") :]
      elif type_name.startswith("__mutated__."):
          return type_name[len("__mutated__.") :]
      else:
          return None
  
imports: from typing import Any, Dict, List, Optional, cast
  from nni.common.framework import get_default_framework
label_desc: Use image node_id 2 for calling the PyTorchOperation obj's underlying member method code with example usage: obj.to_class_name(cls, type_name) and returns: type_name, type_name, None