parameters: (self, input_dataset, batch_size)
docstring: See `Dataset.batch()` for details.
docstring_summary: See `Dataset.batch()` for details.
docstring_tokens: See Dataset . batch () for details .
function: def __init__(self, input_dataset, batch_size):
      """See `Dataset.batch()` for details."""
      super(BatchDataset, self).__init__()
      self._input_dataset = input_dataset
      self._batch_size = ops.convert_to_tensor(batch_size, dtype=dtypes.int64,
                                               name="batch_size")
function_tokens: def __init__ ( self , input_dataset , batch_size ) : super ( BatchDataset , self ) . __init__ ( ) self . _input_dataset = input_dataset self . _batch_size = ops . convert_to_tensor ( batch_size , dtype = dtypes . int64 , name = "batch_size" )