tf.data.experimental.get_structure

Returns the type signature for elements of the input dataset / iterator.

dataset_or_iterator A tf.data.Dataset or an tf.data.Iterator.

A (nested) structure of tf.TypeSpec objects matching the structure of an element of dataset_or_iterator and specifying the type of individual components.

TypeError If input is not a tf.data.Dataset or an tf.data.Iterator object.