预取（tf.data）和prefetch_to_device（tf.contrib）之间的区别

Question

标题本身非常完整，但这里有更多的说明。

prefetch_to_device()预取设备。 它与prefetch有什么不同？预取只是将数据提取到内存吗？

Answer 1

据我所知prefetch_to_device允许您预取数据到GPU，而prefetch则没有。在下面的链接中，您可能会发现像2个月前的@mrry评论。

  

（当前版本的）tf.data API和旧的基于队列的   方法将整个输入管道放在CPU上，而GPU则是   由输入管道之后的图形部分执行。

https://github.com/tensorflow/tensorflow/issues/13610