Linux 拨号vps windows公众号手机端

tensorflow怎么打印模型结构

lewis 5年前 (2020-04-15) 阅读数 11 #大数据
文章标签 tensorflow

在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()

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

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门