code: def plot_trellis_with_path(trellis, path):
      # To plot trellis with path, we take advantage of 'nan' value
      trellis_with_path = trellis.clone()
      for _, p in enumerate(path):
          trellis_with_path[p.time_index, p.token_index] = float("nan")
      plt.imshow(trellis_with_path.T, origin="lower")
      plt.title("The path found by backtracking")
      plt.tight_layout()
  
file_path: audio/examples/tutorials/forced_alignment_tutorial.py
imports: import torch
  import torchaudio
  from dataclasses import dataclass
  import IPython
  import matplotlib.pyplot
label_desc: Use image node_id 5 for calling a global function with example usage: plot_trellis_with_path(trellis, path) without return types