add_layer

最新推荐文章于 2024-01-30 08:46:10 发布

余佩韦君

最新推荐文章于 2024-01-30 08:46:10 发布

阅读量482

点赞数

分类专栏： Tensorflow

本文链接：https://blog.csdn.net/qq_34392093/article/details/78936785

版权

Tensorflow 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

Description

NN网络的神经层添加方法。

Code

def add_layer(inputs, in_size, out_size, activation_function=None):
    Weights = tf.Variable(tf.random_normal([in_size, out_size]))
    biases = tf.Variable(tf.zeros([1, out_size]) + 0.1)
    Wx_plus_b = tf.matmul(inputs, Weights) + biases
    if activation_function is None:
        outputs = Wx_plus_b
    else:
        outputs = activation_function(Wx_plus_b)
    return outputs