tf.nn.softmax

tf.nn.softmax(
    logits,
    axis = None,
    name = None
    dim = None
)

• 作用：softmax函數的作用就是歸一化。
• 輸入：全連接層（往往是模型的最后一層）的值，一般代碼中叫做logits
• 輸出：歸一化的值，含義是屬於該位置的概率，一般代碼叫做probs,例如輸出[0.4, 0.1, 0.2, 0.3],那么這個樣本最可能屬於第0個位置，也就是第0類。這是由於logits的維度大小就設定的任務的類別，所以第0個位置就代表第0類。softmax函數的輸出不改變維度的大小。
• 用途：如果做單分類的問題，那么輸出的值就取top1(最大, argmax)； 如果做多分類問題，那么輸出的值就取topN

例子：

 

 輸出：

 

從結果中分析出array最有可能屬於第0類（一共四類）

 

如果想進一步了解softmax的原理，可以參考這篇文章：https://zhuanlan.zhihu.com/p/93054123