View source on GitHub |
Returns the output shapes for elements of the input dataset / iterator.
tf.compat.v1.data.get_output_shapes(
dataset_or_iterator
)
Migrate to TF2
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.
Description
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.
|