tf.data.experimental.get_structure

TensorFlow 2 version View source on GitHub

Returns the type specification of an element of a Dataset or Iterator.

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

A nested structure of tf.TypeSpec objects matching the structure of an element of dataset_or_iterator and spacifying the type of individal components.

TypeError If dataset_or_iterator is not a Dataset or Iterator object.