code: def _get_batch_shape(data):
      if isinstance(
          data,
          (list, tuple, _tensors.TensorListCPU, _tensors.TensorListGPU),
      ):
          if len(data) == 0:
              return [], True
          if callable(data[0].shape):
              return [x.shape() for x in data], False
          else:
              return [x.shape for x in data], False
      else:
          shape = data.shape
          if callable(shape):
              shape = data.shape()
          return [shape[1:]] * shape[0], True
  
imports: from nvidia.dali import backend
  from nvidia.dali import tensors
  from nvidia.dali import types
  from nvidia.dali._multiproc.messages import TaskArgs, SampleRange
  import nvidia.dali.types
  from nvidia.dali._utils.external_source_impl import get_callback_from_source, accepted_arg_count, SourceKind
label_desc: Use image node_id 1 for calling a global function with example usage: _get_batch_shape(data) and returns: unknown,  True, list,  True, unknown,  False, unknown,  False