1. 程式人生 > >TensorFlow 的softmax實例理解

TensorFlow 的softmax實例理解

col sof .so run tensor class on() 幫助 理論

  對於理論,簡單的去看一下百度上的說明,這裏直接上實例,幫助理解。

1 # softmax函數,將向量映射到0~1的範圍內,P=exp(ax)/(sum(exp(a1x)+exp(a2x)+...))
2 inputdata = tf.Variable([[0.2, 0.1, 0.9]], dtype=np.float32)
3 output = tf.nn.softmax(inputdata)
4 with tf.Session() as sess:
5     sess.run(tf.global_variables_initializer())
6     print (sess.run(inputdata))
7 print (sess.run(output))

  輸出:

[[ 0.2 0.1 0.89999998]]
[[ 0.25519383 0.23090893 0.51389724]]

TensorFlow 的softmax實例理解