pytorch loss function 總結

最近看了下 PyTorch 的損失函數文檔,整理了下本身的理解,從新格式化了公式以下,以便之後查閱。html 值得注意的是,不少的 loss 函數都有 size_average 和 reduce 兩個布爾類型的參數,須要解釋一下。由於通常損失函數都是直接計算 batch 的數據,所以返回的 loss 結果都是維度爲 (batch_size, ) 的向量。python 若是 reduce = Fal
相關文章
相關標籤/搜索