Intellipaat Back

Explore Courses Blog Tutorials Interview Questions
0 votes
2 views
in Data Science by (17.6k points)

Like pandas_df.shape is there any way for tensorflow.data.Dataset? Thanks.

1 Answer

0 votes
by (41.4k points)

If you want to retrieve shapes you can use Dataset._tensors attribute for it as depicted in the below code:

import tensorflow as tf

def dataset_shapes(dataset):

    try:

        return [x.get_shape().as_list() for x in dataset._tensors]

    except TypeError:

        return dataset._tensors.get_shape().as_list()

If you wish to know TensorFlow visit this TensorFlow Tutorial.

31k questions

32.8k answers

501 comments

693 users

Browse Categories

...