code: def __init__(self, module: nn.Module, **kwargs):
      super().__init__()
      self.module = module
      for k, v in kwargs.items():
          setattr(self, k, v)
  
file_path: ColossalAI/colossalai/legacy/nn/layer/colossalai_layer/_utils.py
imports: import torch.nn
  from torch import Tensor
  from ..parallel_2d._operation import split_batch_2d
  from ..parallel_2p5d._operation import split_batch_2p5d
  from ..parallel_3d._operation import split_batch_3d
  from ..utils import get_tensor_parallel_mode
label_desc: Use image node_id 1 to create a new ColossalaiModule object from inherited base classes: nn with example: obj = ColossalaiModule(module)