如何打印输出tensorflow模型
时间: 2023-07-26 20:04:52 浏览: 235
您可以使用`model.summary()`方法来打印输出TensorFlow模型的结构和参数信息。这个方法会给出模型的层次结构,每层的输出形状以及模型参数的数量。
下面是一个简单的例子:
```python
import tensorflow as tf
from tensorflow.keras import layers
# 定义一个简单的模型
model = tf.keras.Sequential([
layers.Dense(64, activation='relu'),
layers.Dense(10, activation='softmax')
])
# 打印模型结构和参数信息
model.summary()
```
运行上面的代码,您将看到类似于以下的输出:
```
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
dense (Dense) (None, 64) 640
_________________________________________________________________
dense_1 (Dense) (None, 10) 650
=================================================================
Total params: 1,290
Trainable params: 1,290
Non-trainable params: 0
_________________________________________________________________
```
这个输出显示了模型的层次结构、每层的输出形状以及模型参数的数量。如果您需要更加详细的信息,可以使用`model.get_config()`方法来获取模型的配置信息。
阅读全文