tensorflow 2.0 中的数据集为 array/list

Question

我是张量流的新手。
我想在 tensorflow 2.0 中直接检查数据集的内容。
我相信在 tensorflow 2.1 中可以做这样的事情：

dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) 
list(dataset.as_numpy_iterator()) 

如何在 tensorflow 2.0 中做同样的事情？

Answer 1

您可以尝试以下方法：

import tensorflow as tf
dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3]) 
[d.numpy() for d in dataset]

Answer 2

由于在 TF2.x 中默认激活急切模式，我更喜欢这种方式：

import tensorflow as tf
dataset = tf.data.Dataset.from_tensor_slices([1, 2, 3])
dataset_it = iter(dataset)
print(dataset_it.__next__()) # output is generated by .numpy()