导入Tensorflow的数据集类

Question

我正在运行Ubuntu 16.04，并已使用pip3 install tensorflow通过pip安装了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？

Answer 1

data已从多个Tensorflow版本中移出contrib。据我所知，最近的更新完全从tf.contrib.data中删除了大多数“标准”操作。现在，它仅包含“实验” /易失性代码。只需使用tf.data.Dataset即可。