如何打印 tf.data 中的一个数据集示例？

Question

我在 tf.data 中有一个数据集。如何轻松打印（或抓取）数据集中的一个元素？

类似于：

print(dataset[0])

Answer 1

在 TF 1.x 中，您可以使用以下内容。提供了不同的迭代器（有些可能在未来的版本中被弃用）。

import tensorflow as tf

d = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4])
diter = d.make_one_shot_iterator()
e1 = diter.get_next()

with tf.Session() as sess:
  print(sess.run(e1))

或在 TF 2.x

import tensorflow as tf

d = tf.data.Dataset.from_tensor_slices([1, 2, 3, 4])
print(next(iter(d)).numpy())

## You can also use loops as follows to traverse the full set one item at a time
for elem in d:
    print(elem)

Answer 2

list(dataset.as_numpy_iterator())[0]

如何打印 tf.data 中的一个数据集示例？

How to print one example of a dataset from tf.data?

python

tensorflow

tensorflow-datasets

tensorflow2.0