code: def test_pytorch():
      setup_pytorch()
      for testcase in [simple_pytorch_op, pytorch_red_channel_op]:
          for device in ["cpu", "gpu"]:
              yield pytorch_case, testcase, device
  
      yield from _gpu_sliced_torch_suite()
      yield from _gpu_permuted_extents_torch_suite()
  
imports: import numpy
  import nvidia.dali.ops
  import nvidia.dali.types
  import os
  import random
  from functools import partial
  from nvidia.dali.pipeline import Pipeline
  from nvidia.dali import fn, pipeline_def
  from nvidia.dali.python_function_plugin import current_dali_stream
label_desc: Use image node_id 12 for calling a global function with example usage: test_pytorch() without return types