如何打印输出用def定义的tensorflow模型

如果您是用`def`定义的TensorFlow模型，您可以使用`tf.keras.utils.plot_model()`方法来打印输出模型的结构图。

以下是一个简单的例子：

import tensorflow as tf
from tensorflow.keras import layers

# 定义一个简单的模型
def my_model():
    inputs = tf.keras.Input(shape=(784,))
    x = layers.Dense(64, activation='relu')(inputs)
    outputs = layers.Dense(10, activation='softmax')(x)
    model = tf.keras.Model(inputs, outputs)
    return model

# 打印模型结构图
model = my_model()
tf.keras.utils.plot_model(model, show_shapes=True)

运行上面的代码，您将看到类似于以下的输出：

Model: "model"
_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
input_1 (InputLayer)         [(None, 784)]             0         
_________________________________________________________________
dense (Dense)                (None, 64)                50240     
_________________________________________________________________
dense_1 (Dense)              (None, 10)                650       
=================================================================
Total params: 50,890
Trainable params: 50,890
Non-trainable params: 0
_________________________________________________________________

这个输出显示了模型的层次结构和每层的输出形状，以及模型参数的数量。如果您需要更多的细节，可以设置`show_layer_names`和`show_shapes`参数来控制显示的内容。