# coding:utf-8

"""
tf_demo06_define_addLayer_for_demo11
Created on 2016/12/9 16:27
@author: GuoYufu
@group : OceanHorn
@contact: OceanHorn@163.com
"""

import tensorflow as tf

def add_layer(inputs, in_size, out_size, n_layer, activation_function=None):

    layer_name = 'layer%s' % n_layer
    with tf.name_scope('weights'):
        weights = tf.Variable(
            initial_value=tf.random_normal(shape=[in_size, out_size])
        )
        tf.histogram_summary(tag=layer_name + '/weights', values=weights)
    with tf.name_scope('biases'):
        biases = tf.Variable(
            tf.zeros([1, out_size]) + 0.1
        )
        tf.histogram_summary(tag=layer_name + '/biases', values=biases)
    with tf.name_scope('layer_output'):
        wx_plus_b = tf.matmul(inputs, weights) + biases

    if activation_function is None:
        outputs = wx_plus_b
    else:
        outputs = activation_function(wx_plus_b)

    tf.histogram_summary(tag=layer_name + '/outputs', values=outputs)
    return outputs