假设我以这种方式定义了一个数据集:
filename_dataset = tf.data.Dataset.list_files("{}/*.png".format(dataset))
Run Code Online (Sandbox Code Playgroud)
如何获取数据集中的元素数量(因此,构成一个纪元的单个元素的数量)?
我知道tf.data.Dataset
已经知道数据集的维度,因为该repeat()
方法允许在指定的时期内重复输入管道。因此,它必须是获取此信息的一种方法。