每班的张量流概率

Question

我希望得到每个类的概率，对于我的输出，在tensorflow中。 使用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>)