對於做分類的網路模型，什麼時候線上性層後面加softmax？以及什麼時候不加，直接線性輸出？

真態-chb2022-01-07 11:16:07

你說的線性層後面沒加softmax，可能只是他的softmax沒有在模型裡而是在損失函數里，比如pytorch裡面的交叉熵函式，是自帶了softmax計算的，所以模型裡面就沒有寫softmax。

修仙2022-01-08 07:20:26

一般做分類，最後都會加上一個softmax的吧

Clarence-XJTU2022-01-09 13:09:21

Softmax直接損失函數里，你沒加的時候網路有的時候預設給你加上了

KodyLee2022-01-14 18:17:53

為什麼轉化成機率，看下交叉熵loss的定義。。。