相关疑难解决方法(0)

tf.data.Dataset:如何获取数据集大小(历元中的元素数)?

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

filename_dataset = tf.data.Dataset.list_files("{}/*.png".format(dataset))
Run Code Online (Sandbox Code Playgroud)

如何获取数据集中的元素数量(因此,构成一个纪元的单个元素的数量)?

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

python-3.x tensorflow tensorflow-datasets

12
推荐指数
9
解决办法
9134
查看次数