Spec-Zone.ru › TensorFlow 2.4
tf.data.experimental.assert_cardinality Утверждает кардинальность входного набора данных.
tf.data.experimental.assert_cardinality(
expected_cardinality
)
Примечание: Предполагается, что «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
Аргументы
expected_cardinality Ожидаемая кардинальность входного набора данных.
Исключения
FailedPreconditionError Утверждение проверяется во время выполнения (при итерировании набора данных) и генерирует ошибку, если фактическая и ожидаемая кардинальность отличаются.