tf.data.experimental.get_structure

View source on GitHub

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

tf.data.experimental.get_structure(
    dataset_or_iterator
)

Args:

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

Returns:

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

Raises:

• TypeError: If dataset_or_iterator is not a Dataset or Iterator object.