def draw_videos_side_by_side(*video_batches: jnp.ndarray,
                             show_num: int = 4):
  """Draws the original and generated videos side-by-side."""
  assert all([v.ndim == 5 for v in video_batches]), 'Incorrect input format.'
  v = np.concatenate(video_batches, 2)
  return draw_videos_grid(v, show_num)
Draws the original and generated videos side-by-side.
videogvt/train_lib/train_utils.py