code: def _get_output_tokens(
      input_metadata: InputMetadata,
  ) -> List[List[int]]:
      output_tokens: List[List[int]] = []
      for seq_group in input_metadata.seq_groups:
          seq_ids, _ = seq_group
          for seq_id in seq_ids:
              seq_data = input_metadata.seq_data[seq_id]
              output_tokens.append(seq_data.output_token_ids)
      return output_tokens
  
imports: from typing import Dict, List, Optional, Tuple
  import torch
  import torch.nn
  from bigdl.llm.vllm.model_executor.input_metadata import InputMetadata
  from bigdl.llm.vllm.sampling_params import SamplingParams, SamplingType
  from bigdl.llm.vllm.sequence import SamplerOutput, SequenceGroupMetadata, SequenceData, SequenceOutputs
  import time
label_desc: Use image node_id 3 for calling a global function with example usage: _get_output_tokens(input_metadata) and returns: output_tokens