tf.train.string_input_producer

创建一个队列

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

例

代码语言：javascript复制

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

代码语言：javascript复制

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)

队列 排序 数据

0 人点赞