Tensorflow 2.0.0: AttributeError: 'TensorSliceDataset' object has no attribute 'as_numpy_iterator'

Question

我正在使用 tensorflow 2.0.0 测试 tensorflow tf.data.Dataset 方法 as_numpy_iterator。根据官方文档https://www.tensorflow.org/api_docs/python/tf/data/Dataset?version=stable#as_numpy_iterator，这个函数允许直接检查tensorflow数据集的内容。但是当我尝试给定的例子时：

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) 
for element in dataset.as_numpy_iterator(): 
  print(element)

发生错误：AttributeError: 'TensorSliceDataset' object has no attribute 'as_numpy_iteractor'。我想知道这个方法是不是新添加的，超出了tensorflow 2.0.0的支持。如果是这样，是否有替代方法将数据集内容检查为 as_numpy_iterator()?

Answer 1

您提供的文档的 link 指向

TensorFlow Core r2.1

将您的 tensorflow 版本更新为 2.1 版本应该可以解决问题；

方法.as_numpy_iterator()在TensorFlow 2.0中没有，只有在TensorFlow >= 2.1

Tensorflow 2.0.0: AttributeError: 'TensorSliceDataset' object has no attribute 'as_numpy_iterator'

Tensorflow 2.0.0: AttributeError: 'TensorSliceDataset' object has no attribute 'as_numpy_iterator'

python

tensorflow

tensorflow2.0

tensorflow2.x