tensorflow probability per class

Question

I want to get probability per class, for my output, in tensorflow. Using softmax yields the following.

A : 0.7
B : 0.2
C : 0.1

But, what I want is

A probability : 0.8
B probability : 0.6
C probability : 0.7

Answer 1

代替使用softmax ，使用tf.nn.sigmoid作为

tf.nn.sigmoid(<output-tensor>)