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

Question

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

Answer 1

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

一种解决方法是对其进行一次迭代以获得元素的数量

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

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