有没有一种简单的方法可以在Tensorflow中获得类似Keras model.summary的东西? [英] Is there an easy way to get something like Keras model.summary in Tensorflow?

问题描述

我一直在和Keras一起工作，真的很喜欢model.summary() 它很好地概述了不同层的大小，尤其是模型具有的参数数量.

I have been working with Keras and really liked the model.summary() It gives a good overview of the size of the different layers and especially an overview of the number of parameters the model has.

Tensorflow中是否有类似的功能?在Stackoverflow或Tensorflow API文档中找不到任何内容.

Is there a similar function in Tensorflow? I could find nothing on Stackoverflow or the Tensorflow API documentation.

推荐答案

我没有看到过类似model.summary()的张量流...但是，我认为您不需要它.有一个TensorBoard，您可以在其中轻松检查NN的体系结构.

I haven't seen anything like model.summary() for the tensorflow... However, I don't think you need it. There is a TensorBoard, where you can easily check the architecture of the NN.

https://www.tensorflow.org/get_started/graph_viz

这篇关于有没有一种简单的方法可以在Tensorflow中获得类似Keras model.summary的东西?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！