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