网络编程
位置：首页>> 网络编程>> Python编程>> 一小时学会TensorFlow2之全连接层

一小时学会TensorFlow2之全连接层

作者：我是小白呀　　发布时间：2022-11-27 10:35:30　

标签：TensorFlow2,全连接层,python

概述

全链接层 (Fully Connected Layer) 会把一个特质空间线性变换到另一个特质空间, 在整个网络中起到分类器的作用.

一小时学会TensorFlow2之全连接层

keras.layers.Dense

keras.layers.Dense可以帮助我们实现全连接.

一小时学会TensorFlow2之全连接层

格式:


tf.keras.layers.Dense(
   units, activation=None, use_bias=True,
   kernel_initializer='glorot_uniform',
   bias_initializer='zeros', kernel_regularizer=None,
   bias_regularizer=None, activity_regularizer=None, kernel_constraint=None,
   bias_constraint=None, **kwargs
)

参数介绍
units正整数, 输出空间维度
activation激活函数, 若不指定, 则不适用激活函数
use_bias布尔值, 该层是否使用偏置向量
kernel_initializerkernel权值矩阵的初始化器
bias_initializer偏执向量的初始化器
kernel_regulaizer运用到偏执项的正则化函数
bias_regularizer运用到偏执项的的正则化函数
activity_regulatizer运用到层的输出正则化函数
kernel_constrint运用到kernel权值矩阵的约束函数
bias_constraint运用到偏执向量的约束函数

例子:


# 创建正态分布
x = tf.random.normal([256, 784])

# 创建全连接层, 输出为512
net = tf.keras.layers.Dense(512)
out = net(x)

# 调试输出
print("w:", net.kernel.shape)
print("b:", net.bias.shape)

输出结果:

w: (784, 512)
b: (512,)

keras.Squential

Squential (序列模型) 是各层次之间依次顺序的线性关系. 模型结构通过一个列表来制定.

一小时学会TensorFlow2之全连接层

格式:


tf.keras.Sequential(
   layers=None, name=None
)

参数:

  • layers: 添加的层

  • name: 数据名称

例子:


# 创建正态分布
x = tf.random.normal([256, 784])

# 建立网络模型
model = tf.keras.Sequential([
   tf.keras.layers.Dense(256, activation="relu"),
   tf.keras.layers.Dense(128, activation="relu"),
   tf.keras.layers.Dense(10, activation="relu"),
])

# 传入x
model(x)

# 调试输出权重和偏置顶名字和形状
for p in model.trainable_variables:
   print(p.name, p.shape)

输出结果:

w: (784, 512)
b: (512,)
dense_1/kernel:0 (784, 256)
dense_1/bias:0 (256,)
dense_2/kernel:0 (256, 128)
dense_2/bias:0 (128,)
dense_3/kernel:0 (128, 10)
dense_3/bias:0 (10,)

来源：https://iamarookie.blog.csdn.net/article/details/117838748

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com