class_docstr: An abstract class that defines a data source.
class_docstr_tok: ["An","abstract","class","that","defines","a","data","source","."]
docstr: Creates a dataset iterator for JAX.
code: def create_iterator(
      self,
      item_ids,
      batch_size: int,
      repeat: bool = True,
      flatten: bool = False,
      shuffle: bool = False,
      prefetch_size: int = 0,
      shuffle_buffer_size: int = 1000000,
      devices: Optional[Sequence[Any]] = None,
  ):
      """Creates a dataset iterator for JAX."""
      dataset = self.create_dataset(
          item_ids,
          flatten=flatten,
          shuffle=shuffle,
          row_shuffle_buffer_size=shuffle_buffer_size,
          pixel_shuffle_buffer_size=shuffle_buffer_size,
      )
      return iterator_from_dataset(
          dataset=dataset,
          batch_size=batch_size,
          repeat=repeat,
          prefetch_size=prefetch_size,
          devices=devices,
      )
  
imports: import abc
  import functools
  import itertools
  from typing import Any, Iterable, Optional, Sequence, Union
  from absl import logging
  from flax import jax_utils
  import jax
  import numpy
  import tensorflow
  from nerfies import camera
  from nerfies import gpath
  from nerfies import image_utils
  from nerfies import tf_camera
  from nerfies import utils
  from tensorflow.python.data.util import nest
label_desc: Use image node_id 24 for calling the DataSource obj's underlying member method code with example usage: obj.create_iterator(item_ids, batch_size, repeat, flatten, shuffle, prefetch_size, shuffle_buffer_size, devices) and returns: iterator_from_dataset