美文网首页
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