code: def concat_aug_pipeline(dev, policy):
      data = types.Constant(np.array([], dtype=np.int32), device=dev)
      if dev == "gpu":
          data = data.gpu()
      data = auto_augment.apply_auto_augment(policy, data)
      return fn.reshape(data, shape=(-1, 2))
  
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 2 for calling a global function with example usage: concat_aug_pipeline(dev, policy) and returns: fn