tf.train.string_input_producer(string_tensor, num_epochs=None, shuffle=True, seed=None, capacity=32, shared_name=None, name=None, cancel_op=None)
See the guide: Inputs and Readers > Input pipeline
Output strings (e.g. filenames) to a queue for an input pipeline.
Note: ifnum_epochs
is notNone
, this function creates local counterepochs
. Uselocal_variables_initializer()
to initialize local variables.
string_tensor
: A 1-D string tensor with the strings to produce.num_epochs
: An integer (optional). If specified, string_input_producer
produces each string from string_tensor
num_epochs
times before generating an OutOfRange
error. If not specified, string_input_producer
can cycle through the strings in string_tensor
an unlimited number of times.shuffle
: Boolean. If true, the strings are randomly shuffled within each epoch.seed
: An integer (optional). Seed used if shuffle == True.capacity
: An integer. Sets the queue capacity.shared_name
: (optional). If set, this queue will be shared under the given name across multiple sessions.name
: A name for the operations (optional).cancel_op
: Cancel op for the queue (optional).A queue with the output strings. A QueueRunner
for the Queue is added to the current Graph
's QUEUE_RUNNER
collection.
ValueError
: If the string_tensor is a null Python list. At runtime, will fail with an assertion if string_tensor becomes a null tensor.Defined in tensorflow/python/training/input.py
.
© 2017 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/string_input_producer