Tensorflow 预取(tf.data)和预取到设备(tf.contrib)之间的差异

Tensorflow 预取(tf.data)和预取到设备(tf.contrib)之间的差异,tensorflow,Tensorflow,标题本身是非常完整的,但这里有更多的澄清 预回迁到设备。 它与其他的有什么不同?预取是否只将数据提取到内存 据我所知,prefetch\u to\u device允许您将数据预取到GPU,而prefetch则不允许。在下面的链接下,你可以找到@mrry两个月前的评论 (当前版本的)tf.data API和旧的基于队列的API 方法将整个输入管道放在CPU上,GPU 由输入管道之后的图形部分执行

标题本身是非常完整的,但这里有更多的澄清

预回迁到设备。
它与其他的有什么不同?预取是否只将数据提取到内存

据我所知,
prefetch\u to\u device
允许您将数据预取到GPU,而
prefetch
则不允许。在下面的链接下,你可以找到@mrry两个月前的评论

(当前版本的)tf.data API和旧的基于队列的API 方法将整个输入管道放在CPU上,GPU 由输入管道之后的图形部分执行