'Args:
ds(DataFlow): the input DataFlow.
batch_size(int): the batch size.
queue (tf.QueueBase): A :class:`tf.QueueBase` whose type
should match the corresponding InputDesc of the model.
Defaults to a FIFO queue of size 3000.'

def __init__(self, ds, batch_size, queue=None):

    super(BatchQueueInput, self).__init__(ds, queue)
    self.batch_size = int(batch_size)
