1. 程式人生 > >pytorch中損失函式的reduce,size_average

pytorch中損失函式的reduce,size_average

size_average是說是不是對一個batch裡面的所有的資料求均值

Reduce size_average result
True True 對batch裡面的資料取均值
True False 對batch裡面的資料求和
False returns a loss per batch element instead,這個時候忽略size_average引數

假設輸入和target的大小分別是NxCxWxH,那麼一旦reduce設定為False,loss的大小為NxCxWxH,返回每一個元素的loss

reduction代表了上面的reduce和size_average雙重含義,這也是文件裡為什麼說reduce和size_average要被Deprecated 的原因