Returns the output shapes of a Dataset
or Iterator
elements.
tf.data.get_output_shapes( dataset_or_iterator )
This utility method replaces the deprecated-in-V2 tf.compat.v1.Dataset.output_shapes
property.
Args | |
---|---|
dataset_or_iterator | A tf.data.Dataset or tf.data.Iterator . |
Returns | |
---|---|
A nested structure of tf.TensorShape objects matching the structure of the 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_shapes