为什么我为 tf.data.Dataset.from_tensor_slices(X) 获取 DatasetV1Adapter 返回类型而不是 TensorSliceDataset

Question

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

import tensorflow as tf

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

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

Answer 1

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

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