code: def sequence_batch_output_size(
      unfolded_extents, input_batch, angle_batch, axis_batch=None
  ):
      def iter_by_groups():
          assert sum(unfolded_extents) == len(input_batch)
          assert len(input_batch) == len(angle_batch)
          assert axis_batch is None or len(axis_batch) == len(
              angle_batch
          )
          offset = 0
          for group in unfolded_extents:
              yield input_batch[offset : offset + group], angle_batch[
                  offset : offset + group
              ], None if axis_batch is None else axis_batch[
                  offset : offset + group
              ]
              offset += group
  
      sequence_output_shape = [
          infer_sequence_size(
              [frame.shape for frame in input_frames], angles, axes
          )
          for input_frames, angles, axes in iter_by_groups()
      ]
      return [
          output_shape
          for output_shape, num_frames in zip(
              sequence_output_shape, unfolded_extents
          )
          for _ in range(num_frames)
      ]
  
imports: import cv2
  import numpy
  import math
  import os
  import random
  from nvidia.dali.pipeline import Pipeline
  import nvidia.dali.ops
  import nvidia.dali.types
  import nvidia.dali
  from test_utils import compare_pipelines
  from sequences_test_utils import ArgData, ArgDesc, ArgCb, ParamsProvider, get_video_input_cases, sequence_suite_helper
label_desc: Use image node_id 14 for calling a global function with example usage: sequence_batch_output_size(unfolded_extents, input_batch, angle_batch, axis_batch) and returns: unknown