python如何打印输出def函数定义的tensorflow网络模型

打印输出 TensorFlow 网络模型的定义可以使用 `tf.keras.utils.plot_model` 函数，该函数可以将模型的结构以图的形式展示出来，方便我们查看模型的结构。

具体操作如下：

1. 导入 TensorFlow 和相关库：

import tensorflow as tf
from tensorflow.keras.utils import plot_model

2. 定义 TensorFlow 网络模型：

def my_model():
    model = tf.keras.Sequential([
        tf.keras.layers.Input(shape=(28, 28, 1)),
        tf.keras.layers.Conv2D(32, 3, activation='relu'),
        tf.keras.layers.MaxPooling2D(),
        tf.keras.layers.Flatten(),
        tf.keras.layers.Dense(10, activation='softmax')
    ])
    return model

3. 使用 `plot_model` 函数，将模型的结构以图的形式展示出来，并保存到本地：

model = my_model()
plot_model(model, to_file='my_model.png', show_shapes=True)

执行上述代码后，会在当前目录下生成一个名为 `my_model.png` 的文件，打开该文件即可查看模型的结构图。