tensorflow2.0 简单搭建4层全连接神经网络

最新推荐文章于 2023-01-17 23:35:19 发布

小贝学cs

最新推荐文章于 2023-01-17 23:35:19 发布

阅读量2.6k

点赞数 3

分类专栏： tensorflow2.0 实战

本文链接：https://blog.youkuaiyun.com/qq_40318502/article/details/106534187

版权

tensorflow2.0 实战专栏收录该内容

3 篇文章

订阅专栏

利用tf的框架搭建一个如图的神经网络
在这里插入图片描述

import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers,optimizers,losses,datasets,Sequential

model=keras.Sequential([
    layers.Dense(256,activation='relu'),
    layers.Dense(256,activation='relu'),
    layers.Dense(256,activation='relu'),
    layers.Dense(10)
])

model.build(input_shape=(4,784))#模型的搭建
model.summary()#查看每一层的信息

#。以第一层为例，输入特征长度为 784，输 出特征长度为 256，当前层的参数量为 784 ∙ 256 + 256 = 200960 ，
# 同样的方法可以计算第 二、三、四层的参数量.