読者です 読者をやめる 読者になる 読者になる

キンサクプログラマー

お金儲けと技術のメモ

tensorflowでsoftmax

tensorflowのtutorialでよく使われてるsoftmax関数。
説明はこことかにのってた
ソフトマックス関数 - 機械学習の「朱鷺の杜Wiki」

テンソルの和が1になるように正規化して出力するだけっぽい。
 

import tensorflow as tf
x = tf.placeholder(tf.float32,[3,None])

W = tf.placeholder(tf.float32,[4,3])
#W = tf.Variable([8,4]))
debug=[[0.],[1.],[2.]]
weight=[[5.,6.,7.],
          [5.,6.,7.],
          [5.,6.,7.],
          [5.,6.,7.]]

b =  [0.,0.,0.,0.]

rs = tf.nn.softmax(tf.matmul(W,x)+b)
with tf.Session() as sess:
    result = sess.run(rs,feed_dict={W:weight,x:debug})
    print result

with tf.Session() as sess:
    result = sess.run(tf.matmul(weight,debug))
    print result