tf.TFRecordReader.read_up_to()

tf.TFRecordReader.read_up_to(queue, num_records, name=None)

Returns up to num_records (key, value pairs) produced by a reader.

Will dequeue a work unit from queue if necessary (e.g., when the Reader needs to start reading from a new file since it has finished with the previous file). It may return less than num_records even before the last batch.

Args:

• queue: A Queue or a mutable string Tensor representing a handle to a Queue, with string work items.
• num_records: Number of records to read.
• name: A name for the operation (optional).

Returns:

A tuple of Tensors (keys, values).

• keys: A 1-D string Tensor.
• values: A 1-D string Tensor.