Ram Sure
Ram Sure

Reputation: 83

Display Tensorflow Model Summary as like in Keras

We can build the model with tensorflow layers. Is there any way we can display the model summary as like in Keras.

Keras Model Summary

Upvotes: 5

Views: 5645

Answers (2)

prosti
prosti

Reputation: 46351

Keras is part of TensorFlow (for some time) so you can always get nice things like:

model.output_shape # model summary representation
model.summary() # model configuration
model.get_config() # list all weight tensors in the model
model.get_weights() # get weights and biases

Upvotes: -2

Zvika
Zvika

Reputation: 1280

No, there is no such option. TensorFlow is a lot more generic than Keras and allows arbitrary graph architectures, so showing such a structured summary does not make sense for arbitrary TensorFlow graphs. The closest is probably TensorBoard, which has a very handy interactive graph visualization tool.

Upvotes: 3

Related Questions