python - 导入 TensorFlow 的 Dataset 类

Question

我正在运行 Ubuntu 16.04，并通过 pip 安装了 tensorflow，使用pip3 install tensorflow. 我现在想导入和使用Dataset该类，它是contrib.data包的一部分。在各种教程中，这只是使用：

import tensorflow as tf
from tf.contrib.data import Dataset

但这打印出来：

ImportError: No module named 'tf'

我也尝试过在不重命名 tensorflow 的情况下使用它：

import tensorflow
from tensorflow.contrib.data import Dataset

但这打印出来：

ImportError: cannot import name 'Dataset'

如果我导航到/usr/local/lib/python3.5/dist-packages/tensorflow/contrib/data，则有两个目录（__pycache__和python）和两个文件（_dataset_ops.so, __init__.py）。但我不确定Dataset应该在哪里定义类，如果有的话......

那么我应该怎么做才能让我导入Dataset呢？

标签： pythontensorflowtensorflow-datasets

Answer 1

datacontrib之前已从几个 TensorFlow 版本中移出。据我所知，最近的更新完全从tf.contrib.data. 它现在只包含“实验性”/易失性代码。只需使用即可tf.data.Dataset。