code: def collect_sub_policy_outputs(sub_policies, num_magnitude_bins):
      sub_policy_outputs = []
      for sub_policy in sub_policies:
          out = []
          for aug, _, mag_bin in sub_policy:
              magnitudes = aug._get_magnitudes(num_magnitude_bins)
              param = aug._map_mag_to_param(magnitudes[mag_bin])
              out.append(param)
          sub_policy_outputs.append(out)
      return sub_policy_outputs
  
imports: import itertools
  import os
  import unittest
  import random
  import numpy
  from scipy.stats import chisquare
  from nose2.tools import params
  from nvidia.dali import fn, types
  from nvidia.dali import pipeline_def
  from nvidia.dali.auto_aug import auto_augment, augmentations
  from nvidia.dali.auto_aug.core import augmentation, Policy
  from test_utils import get_dali_extra_path, check_batch
  from nose_utils import assert_raises, assert_warns
label_desc: Use image node_id 3 for calling a global function with example usage: collect_sub_policy_outputs(sub_policies, num_magnitude_bins) and returns: sub_policy_outputs