Python tf.data.TFRecordDataset.cardinality用法及代码示例

用法

cardinality()

返回数据集的基数(如果已知)。

如果数据集包含无限数量的元素，cardinality 可能会返回 tf.data.INFINITE_CARDINALITY，如果分析无法确定数据集中元素的数量(例如，当数据集源是文件时)，则可能会返回 tf.data.UNKNOWN_CARDINALITY。

dataset = tf.data.Dataset.range(42)
print(dataset.cardinality().numpy())
42
dataset = dataset.repeat()
cardinality = dataset.cardinality()
print((cardinality == tf.data.INFINITE_CARDINALITY).numpy())
True
dataset = dataset.filter(lambda x:True)
cardinality = dataset.cardinality()
print((cardinality == tf.data.UNKNOWN_CARDINALITY).numpy())
True