如何找到张量流数据集对象的大小？

Question

我已经创建了tensorflow数据集对象，我想知道这个数据集的大小。

Answer 1

tf.data.Dataset的长度没有完全定义。

一种解决方法是将其迭代一次以获取元素数量

def get_ds_length(dataset):
    len = 0
    for _ in dataset:
        len += 1
    return len

显然，这对于大型数据集和使用大量预处理的数据集来说会很慢。