code: def iterator_function_def(shard_id=0, num_shards=1):
      files, labels = fn.readers.file(
          name="reader",
          files=file_names,
          labels=file_labels,
          shard_id=shard_id,
          num_shards=num_shards,
      )
  
      return labels.gpu()
  
file_path: DALI/dali/test/python/jax_plugin/utils.py
imports: import os
  import numpy
  from nvidia.dali import pipeline_def
  from nvidia.dali.backend import TensorGPU
  import nvidia.dali.fn
  import nvidia.dali.types
label_desc: Use image node_id 5 for calling a global function with example usage: iterator_function_def(shard_id, num_shards) and returns: labels