tensorflow常用函数

最新推荐文章于 2023-12-20 18:19:19 发布

原创最新推荐文章于 2023-12-20 18:19:19 发布 · 115 阅读

0 ·

CC 4.0 BY-SA版权

tensorflow 专栏收录该内容

1 篇文章

订阅专栏

本文深入解析了tf.layers.dense函数的参数及其用法，包括输入、输出维度、激活函数、偏置等关键属性，并介绍了如何使用slim库设置默认的步长和填充方式。

tf.layers.dense
tf.layers.dense(
inputs,
units,
activation=None,
use_bias=True,
kernel_initializer=None,
bias_initializer=tf.zeros_initializer(),
kernel_regularizer=None,
bias_regularizer=None,
activity_regularizer=None,
kernel_constraint=None,
bias_constraint=None,
trainable=True,
name=None,
reuse=None
)

Functional interface for the densely-connected layer紧密连接层的功能接口

inputs:此层的输入

units：此层的输出维度

activation=None, 激活函数
use_bias=True,偏置

with slim.arg_scope([slim.conv2d, slim.max_pool2d, slim.avg_pool2d],
stride=1, padding='SAME'):

表示slim.conv2d, slim.max_pool2d, slim.avg_pool2d函数的stride=1, padding='SAME'为默认值，这样减少代码的抒写