code: def plot():
      fig, ax = plt.subplots(3, 1)
      for i in range(3):
          with torch.inference_mode():
              spec, spec_lengths, _ = tacotron2.infer(
                  processed, lengths
              )
          print(spec[0].shape)
          ax[i].imshow(
              spec[0].cpu().detach(), origin="lower", aspect="auto"
          )
  
imports: import torch
  import torchaudio
  import IPython
  import matplotlib.pyplot
label_desc: Use image node_id 2 for calling a global function with example usage: plot() without return types