张量流中shuffle_batch()和batch()之间的区别

时间:2017-07-14 07:21:25

标签: tensorflow records

我使用tf.train.shuffle_batch()tf.train.batch()来读取.record文件,每个类有两个类和三个示例。如果我想要两批三个例子,我将num_epochs = 1放在string_input_producer中,这样我就可以得到两批三个例子,但每次调用它们时,两个函数都会返回两个洗牌批次。 / p>

这两个功能有什么区别?

谢谢!

1 个答案:

答案 0 :(得分:1)

tf._train.shuffle_batch()会以随机顺序返回记录,而tf.train.batch()会按顺序从您的来源返回记录。根据您要完成的任务,一个或另一个可能更有帮助。我会说shuffle_batch,因为它为学习过程增加了额外的随机性,但首先要考虑你的任务是否有意义。