Tensorflow学习笔记(5)-网络结构的构建

1.构建网络结构

# -*- coding: utf-8 -*-
# @Time    : 17-8-4 上午10:57
# @Author  : 未来战士biubiu！！
# @FileName: 5-buildstructure.py
# @Software: PyCharm Community Edition
# @Blog    ：http://blog.csdn.net/u010105243/article/
from __future__ import print_function
import tensorflow as tf
import numpy as np

# 自定义网络的结构
def layer(inputs, in_size, out_size, activation_function=None):
    # 定义网络的权重，随机分配初始值效果更好，这里采用均匀分布，也可以替换其他的
    # [in_size,out_size]表示的是该层网络结构
    W = tf.Variable(tf.random_uniform([in_size, out_size]))
    b = tf.Variable(tf.zeros([1, out_size]) + 0.1)
    y_ = tf.matmul(inputs, W) + b  # W*x+b
    if activation_function