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.

Welcome to Intellipaat Community. Get your technical queries answered by top developers!

30.5k questions

32.5k answers

500 comments

108k users

Browse Categories

...