tensorflow2打印模型结构

TensorFlow 2 中，你可以使用 `tf.keras` API 来方便地查看模型的结构。一旦创建了模型实例，可以调用 `model.summary()` 函数来打印出模型的详细信息，包括每个层的名称、输入和输出维度、以及参数的数量。

例如：

import tensorflow as tf
from tensorflow import keras

# 创建一个简单的模型
model = keras.Sequential([
    keras.layers.Dense(64, activation='relu', input_shape=(784,)),
    keras.layers.Dense(10)
])

# 打印模型结构
model.summary()

这将输出类似这样的结果：

Model: "sequential"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
dense (Dense)                (None, 64)                5120      
_________________________________________________________________
dense_1 (Dense)              (None, 10)                650       
=================================================================
Total params: 5,770
Trainable params: 5,770
Non-trainable params: 0
_________________________________________________________________