tf.train.string_input_producer

Posted aofengdaxia

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了tf.train.string_input_producer相关的知识,希望对你有一定的参考价值。

tf.train.string_input_producer(
    string_tensor,
    num_epochs=None,
    shuffle=True,
    seed=None,
    capacity=32,
    shared_name=None,
    name=None,
    cancel_op=None
)

把输入的数据进行按照要求排序成一个队列。最常见的是把一堆文件名整理成一个队列

filenames = [os.path.join(data_dir,'data_batch%d.bin' % i ) for i in xrange(1,6)]
filename_queue = tf.train.string_input_producer(filenames)

以上是关于tf.train.string_input_producer的主要内容,如果未能解决你的问题,请参考以下文章