tf.data.get_output_types

Returns the output shapes of a Dataset or Iterator elements.

This utility method replaces the deprecated-in-V2 tf.compat.v1.Dataset.output_types property.

Args
dataset_or_iterator A tf.data.Dataset or tf.data.Iterator.
Returns
A nested structure of tf.DType objects objects matching the structure of dataset / iterator elements and specifying the shape of the individual components.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/data/get_output_types