Spec-Zone.ru › TensorFlow
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 Проверка выполняется во время выполнения (при переборе набора данных); если фактическое и ожидаемое количество элементов различается, возникает ошибка.