3.添加神经层

'''
Author: 365JHWZGo
Description: 3.添加神经层
Date: 2021-10-12 12:05:55
FilePath: \tensorflow\tensorflow\day02-4.py
LastEditTime: 2021-10-12 14:38:17
LastEditors: 365JHWZGo
'''
import tensorflow as tf

# 定义添加神经层的函数
def add_layer(inputs, in_size, out_size, activaton_function=None):
    '''
    inputs 输入
    #in_size 输入大小
    #out_size 输出大小
    #activation_function 激励函数
    '''
    # Weights 权重，生成in_size行out_size列的矩阵
    Weights = tf.Variable(tf.random_normal([in_size, out_size]))

    # biases 偏置，不推荐为0
    biases = tf.Variable(tf.zeros([1, out_size])+0.1)

    # 定义Wx_plus_b为神经网络未激活的值
    Wx_plus_b = tf.matmul(inputs, Weights)+biases

    #激励函数
    if activaton_function is None:
        outputs = Wx_plus_b
    else:
        outputs = activaton_function(Wx_plus_b)

    #返回结果
    return outputs