module'tensorflow_core._api.v2.data'没有属性'Iterator'

Question

无法弄清楚使用什么代替迭代器

我改用了tf.compat.v1.data.Iterator，但又遇到了另一个错误-AttributeError: 'PrefetchDataset' object has no attribute 'output_types'

代码：

train_ds = prepare_for_train(labeled_ds)

val_ds = tf.data.Dataset.from_tensor_slices(test_data)

#create a iterator with shape and type

iter = tf.data.Iterator.from_structure(train_ds.output_types, train_ds.output_shapes)

"""iter= tf.compat.v1.data.Iterator.from_structure(train_ds.output_types, train_ds.output_shapes)"""
print(iter)


*AttributeError: module 'tensorflow_core._api.v2.data' has no attribute 'Iterator'*

我的T​​F版本2.2.0-dev20200212

谢谢！

Answer 1

我能够重现您的错误。这是在Tensorflow Version 2.x中修复它的方法。