Loss總結

1. BCELoss (BCELoss)BCEWithLogitsLoss用於單標籤二分類或者多標籤二分類，輸出和目標的維度是(batch,C)，batch是樣本數量，C是類別數量，對於每一個batch的C個值，對每個值求sigmoid到0-1之間，所以每個batch的C個值之間是沒有關係的,相互獨立的，所以之和不一定爲1。每個C值代表屬於一類標籤的概率。如果是單標籤二分類，那輸出和目標的維度是(

>>阅读原文<<