TensorFlow数据读取方式:Dataset API
英文详细版参考: https://www.cnblogs.com/jins-note/p/10243716.html Dataset API是TensorFlow 1.3版本中引入的一个新的模块,主要服务于数据读取,构建输入数据的pipeline。 此前,在TensorFlow中读取数据一般有两种方法: 使用placeholder读内存中的数据 使用queue读硬盘中的数据(关于这种方式,可以参考我之前的一篇文章: 十图详解tensorflow数据读取机制 ) 相Dataset API同时支持从内存和硬盘的读取,相比之前的两种方法在语法上 更加简洁易懂。此外,如果想要用到TensorFlow新出的Eager模式,就必须要使用Dataset API来读取数据。 本文就来为大家详细地介绍一下Dataset API的使用方法(包括在非Eager模式和Eager模式下两种情况)。 Dataset API的导入 在TensorFlow 1.3中,Dataset API是放在contrib包中的: tf . contrib . data . Dataset 而在TensorFlow 1.4中,Dataset API已经从contrib包中移除,变成了核心API的一员: tf . data . Dataset 下面的示例代码将以TensorFlow 1.4版本为例,如果使用TensorFlow