python - 为什么我得到 DatasetV1Adapter 返回类型而不是 TensorSliceDataset for tf.data.Dataset.from_tensor_slices(X)

Question

我只是在关注“使用 scikit-learn 和 tensorflow 进行机器学习”一书中的代码示例。

import tensorflow as tf

X = tf.range(10)
dataset = tf.data.Dataset.from_tensor_slices(X)

根据这本书，我应该得到变量“数据集”的类型“TensorSliceDataset shapes:(), types: tf.int32”，但我得到的是“DatasetV1Adapter shapes: (), types: tf.int32”

Answer 1

根据他们的文档，如果您使用的是 tf 2.0(或更低版本)，它不支持 TensorSliceDataset，并且会给您 DatasetV1Adapter https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/raw_ops

您将需要 TF 2.1.x 及更高版本