美文网首页
2018-04-01TensorFlow初步,神经网络的一层

2018-04-01TensorFlow初步,神经网络的一层

作者: DIO哒 | 来源:发表于2018-04-01 15:45 被阅读0次
    import tensorflow as tf
    
    def add_layer(inputs,in_size,out_size,activation_function=None):
        Weights = tf.Variable(tf.random_normal([in_size,out_size]))#row * col
        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
    

    图示的话大概是下面这样


    image.png

    相关文章

      网友评论

          本文标题:2018-04-01TensorFlow初步,神经网络的一层

          本文链接:https://www.haomeiwen.com/subject/ivxacftx.html