首先先交代結論:nll_loss()與CrossEntropyLoss()損失函式計算的關係為CrossEntropyLoss()等於對輸入資料先做softmax,再做log處理,再加nll_loss()操作。
確定! 回上一頁