当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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


断言输入数据集的基数。

用法

tf.data.experimental.assert_cardinality(
    expected_cardinality
)

参数

  • expected_cardinality 输入数据集的预期基数。

返回

抛出

  • 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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。