python - tf.data.Dataset：如何获取数据集大小（一个纪元中的元素数）？

Question

假设我以这种方式定义了一个数据集：

filename_dataset = tf.data.Dataset.list_files("{}/*.png".format(dataset))

如何获取数据集中的元素数量（因此，构成一个时期的单个元素的数量）？

我知道tf.data.Dataset已经知道数据集的维度，因为该repeat()方法允许将输入管道重复指定数量的时期。所以它必须是一种获取这些信息的方法。

标签： pythonpython-3.xtensorflowtensorflow-datasets

Answer 1

len(list(dataset))在渴望模式下工作，尽管这显然不是一个好的通用解决方案。