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 [ ]: