code: def print_token_outputs(ids: List[int], *xs: torch.Tensor):
      """
      ### Print tokens from model outputs
  
      Pretty prints target tokens along side outputs from the model(s).
  
      :param ids: are the target token ids
      :param xs: are the model(s) outputs
      """
      ids = ids + [-1]
      xs = [[-1] + x[0].max(dim=-1)[1].tolist() for x in xs]
  
      print_tokens(ids, xs)
  
file_path: annotated_deep_learning_paper_implementations/labml_nn/neox/utils/__init__.py
imports: import typing
  from typing import List, Optional
  import torch
  from labml import logger
  from labml.logger import Text
  from labml_nn.neox.tokenizer import get_tokenizer
label_desc: Use image node_id 2 for calling a global function with example usage: print_token_outputs(ids) without return types