code: def infer_sequence_size(input_shapes, angles, axes=None):
      assert len(input_shapes) == len(angles)
      assert axes is None or len(axes) == len(angles)
      if axes is None:
          no_correction_shapes = [
              np.array(
                  get_output_size(math.radians(angle), shape, False),
                  dtype=np.int32,
              )
              for shape, angle in zip(input_shapes, angles)
          ]
          corrected_shapes = [
              np.array(
                  get_output_size(math.radians(angle), shape, True),
                  dtype=np.int32,
              )
              for shape, angle in zip(input_shapes, angles)
          ]
      else:
          no_correction_shapes = [
              np.array(
                  get_3d_output_size(
                      math.radians(angle), axis, shape, False
                  ),
                  dtype=np.int32,
              )
              for shape, angle, axis in zip(input_shapes, angles, axes)
          ]
          corrected_shapes = [
              np.array(
                  get_3d_output_size(
                      math.radians(angle), axis, shape, True
                  ),
                  dtype=np.int32,
              )
              for shape, angle, axis in zip(input_shapes, angles, axes)
          ]
      max_shape = np.max(no_correction_shapes, axis=0)
      parity = np.sum(
          np.array(corrected_shapes, dtype=np.int32) % 2, axis=0
      )
      for i in range(len(max_shape)):
          if max_shape[i] % 2 != (2 * parity[i] > len(input_shapes)):
              max_shape[i] += 1
      return max_shape
  
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 13 for calling a global function with example usage: infer_sequence_size(input_shapes, angles, axes) and returns: max_shape