code: def test_stack():
      for num_inputs in [1, 2, 3, 100]:
          for layout, ndim in [
              (None, 0),
              (None, 1),
              ("A", 1),
              (None, 2),
              ("HW", 2),
              (None, 3),
              ("DHW", 3),
          ]:
              for axis in range(-ndim, ndim + 1):
                  axis_names = (
                      [None]
                      if layout is None and ndim > 0
                      else [None, "C"]
                  )
                  for axis_name in axis_names:
                      yield _run_test_stack, num_inputs, layout, ndim, axis, axis_name
  
file_path: DALI/dali/test/python/operator_1/test_join.py
imports: import nvidia.dali
  import nvidia.dali.fn
  import numpy
  import math
  from test_utils import check_batch
label_desc: Use image node_id 10 for calling a global function with example usage: test_stack() without return types