tensorflow怎么打印模型结构

1085
2024/2/23 20:41:24
栏目: 深度学习
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

在TensorFlow中,可以使用model.summary()方法来打印模型的结构。下面是一个示例:

import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense

# 创建一个简单的神经网络模型
model = Sequential([
    Dense(64, activation='relu', input_shape=(784,)),
    Dense(10, activation='softmax')
])

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

运行上面的代码,就可以打印出神经网络模型的结构信息,包括每一层的名称、类型、输出形状和参数数量等。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: TensorFlow中的图像分类任务中如何应用迁移学习