tf.data.experimental.assert_cardinality

Asserts the cardinality of the input dataset.

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.data.experimental.assert_cardinality

 
tf.data.experimental.assert_cardinality(
    expected_cardinality
)

Used in the notebooks

Used in the tutorials 
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 The expected cardinality of the input dataset.

A Dataset transformation function, which can be passed to tf.data.Dataset.apply.

FailedPreconditionError The assertion is checked at runtime (when iterating the dataset) and an error is raised if the actual and expected cardinality differ.