class_docstr: OutputGenerator is a generic class to generate strategies for Output Node.
code: def __init__(
      self,
      operation_data_mapping: Dict[str, OperationData],
      device_mesh: DeviceMesh,
      predecessor_nodes: List[Node],
      output_option: str,
  ):
      super().__init__(
          operation_data_mapping, device_mesh, predecessor_nodes
      )
      self.output_option = output_option
  
file_path: colossalai/colossalai/auto_parallel/tensor_shard/node_handler/strategy/output_generator.py
imports: from typing import Dict, List
  from torch.fx import Node
  from colossalai.auto_parallel.tensor_shard.sharding_strategy import MemoryCost, OperationData, ShardingStrategy, TrainCycleItem
  from colossalai.device.device_mesh import DeviceMesh
  from .strategy_generator import OutputStrategyGenerator
label_desc: Use image node_id 1 to create a new OutputGenerator object from inherited base classes: OutputStrategyGenerator with example: obj = OutputGenerator(operation_data_mapping, device_mesh, predecessor_nodes, output_option)