code: def __init__(self, inputs, batch_size, extra_inputs=None):
      self._inputs = [i for i in inputs]
      if extra_inputs is None:
          extra_inputs = []
      self._extra_inputs = extra_inputs
      self._batch_size = batch_size
      if batch_size is not None:
          self._ids = np.arange(self._inputs[0].shape[0])
          self.update()
  
imports: import numpy
label_desc: Use image node_id 1 to create a new BatchDataset object  with example: obj = BatchDataset(inputs, batch_size, extra_inputs)