每类的张量流概率

Question

我想获得每个类的概率，以张量流为单位。 使用softmax会产生以下结果。

A : 0.7
B : 0.2
C : 0.1

但是，我想要的是

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

Answer 1

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

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