This is a legacy API for inspecting the type signature of dataset elements. In
TF 2, you should use the tf.data.Dataset.element_spec attribute instead.
A (nested) structure of tf.TensorShape objects matching the structure of
the dataset / iterator elements and specifying the shape of the individual
components.