In [1]:

    

import tensorflow as tf

softmax_data = [0.7, 0.2, 0.1]
one_hot_data = [1.0, 0.0, 0.0]

softmax = tf.placeholder(tf.float32)
one_hot = tf.placeholder(tf.float32)


    

In [4]:

    

xEntropy = -tf.reduce_sum( tf.multiply(one_hot, tf.log(softmax)))

with tf.Session() as sess:
    output = sess.run(xEntropy, feed_dict={softmax: softmax_data, one_hot: one_hot_data})

print(output)


    

    




0.356675

In [ ]: