如何使用TensorFlow中的Dataset API（使用内置输入管道，告别‘feed-dict’ ）

Posted 2021-04-24 AI科技大本营

翻译 | AI科技大本营

参与 | zzq

审校 | reason_W

 

本文已更新至TensorFlow1.5版本

 

我们知道，在TensorFlow中可以使用feed-dict的方式输入数据信息，但是这种方法的速度是最慢的，在实际应用中应该尽量避免这种方法。而使用输入管道就可以保证GPU在工作时无需等待新的数据输入，这才是正确的方法。

幸运的是，TensorFlow提供了一种内置的API——Dataset，使得我们可以很容易地就利用输入管道的方式输入数据。在这篇教程中，我们将介绍如何创建和使用输入管道以及如何高效地向模型输入数据。

这篇文章将解释DatasetAPI的基本工作机制，并给出了几种最常用的例子。

 

https://github.com/FrancescoSaverioZuppichini/Tensorflow-Dataset-Tutorial/blob/master/dataset_tutorial.ipynb

▌