當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.data.experimental.assert_cardinality用法及代碼示例

斷言輸入數據集的基數。

用法

tf.data.experimental.assert_cardinality(
    expected_cardinality
)

參數

expected_cardinality 輸入數據集的預期基數。

返回

一個 Dataset 轉換函數，可以傳遞給 tf.data.Dataset.apply 。

拋出

FailedPreconditionError 在運行時(迭代數據集時)檢查斷言，如果實際和預期的基數不同，則會引發錯誤。

注意：以下假設 "examples.tfrecord" 包含 42 條記錄。

dataset = tf.data.TFRecordDataset("examples.tfrecord")
cardinality = tf.data.experimental.cardinality(dataset)
print((cardinality == tf.data.experimental.UNKNOWN_CARDINALITY).numpy())
True
dataset = dataset.apply(tf.data.experimental.assert_cardinality(42))
print(tf.data.experimental.cardinality(dataset).numpy())
42

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.data.experimental.assert_cardinality。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。