美文网首页
Tensorlayer神经网络基本定义结构

Tensorlayer神经网络基本定义结构

作者: yalesaleng | 来源:发表于2018-07-19 15:01 被阅读21次
    • 首先:
    network = tf.layers.InputLayer(x, name = 'input_layer')
    
    • 然后:
    conv:
         network = .....
    fc:
         network = FlattenLayer(network, name='flatten')
         network = DenseLayer(network, n_units=1024, act=tf.nn.relu, name='fc1')
         network = DenseLayer(network, n_units=1024, act=tf.nn.relu, name='fc2')
         network = DenseLayer(network, n_units=2, act=tl.activation.identity, name='fc3')
    

    最后:

    y = network.outputs
    logit = tf.nn.softmax(y)
    
    输出预测logit值
    

    相关文章

      网友评论

          本文标题:Tensorlayer神经网络基本定义结构

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