无法将tf.data.Dataset对象转换为numpy迭代器

Question

我正在使用Tensorflow 1.14.0和tensorflow_datasets 1.2.0

尝试运行以下代码时

import tensorflow as tf
import tensorflow_datasets as tfds

smallnorb = tfds.load("smallnorb")
smallnorb_train, smallnorb_test = smallnorb["train"], smallnorb["test"]
assert isinstance(smallnorb_train, tf.data.Dataset)
smallnorb_train = smallnorb_train.as_numpy_iterator()

我收到以下错误

AttributeError: 'DatasetV1Adapter' object has no attribute 'as_numpy_iterator'

根据tensorflow_datasets docs，这应该起作用。

为什么不呢？为什么首先要获得一个DatasetV1Adapter对象？

Answer 1

除非出于某些特殊原因需要2.x，否则请使用1.x。

如果使用tensorflow 2.1.0和tensorflow_datasets 2.0.0，此代码有效。可以找到1.x的tf.data.Dataset的正确文档，但实际上没有这种方法。