code: def _convert(output):
      if isinstance(output, torch.Tensor) and not isinstance(
          output, ColoTensor
      ):
          output.__class__ = ColoTensor
      elif isinstance(output, (list, tuple)):
          output = type(output)(_convert(o) for o in output)
      return output
  
file_path: ColossalAI/colossalai/tensor/colo_tensor.py
imports: from functools import lru_cache
  from typing import Callable, Set
  import torch
label_desc: Use image node_id 2 for calling a global function with example usage: _convert(output) and returns: output