Python搭建tensorflow三层神经网络

Python搭建tensorflow三层神经网络的小例子

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt #输出图像的库

#加入神经网络层，activation_function是激励函数，初始化为None
def add_layer(inputs,in_size,out_size,activation_function=None):

    #定义权重(矩阵),为一个有in_size行，out_size列的矩阵,矩阵元素值取自一个正态分布中的随机数。
    Weights=tf.Variable(tf.random_normal([in_size,out_size]))#random_normal

    #定义偏置(列表)，初始值推荐不为零
    biases=tf.Variable(tf.zeros([1,out_size]))+0.1

    #Weight*x+biases
    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 outpu