L1Loss¶
- class torch.nn.L1Loss(size_average=None, reduce=None, reduction='mean')[源][源]¶
建立一個準則,用於測量輸入 和目標 中每個元素之間的平均絕對誤差 (MAE)。
未進行歸約的損失(即,將
reduction設定為'none')可以描述為其中 是批次大小。如果
reduction不是'none'(預設為'mean'),則和 是任意形狀的 Tensor,每個共有 個元素。
求和操作仍然對所有元素進行,併除以 。
如果將
reduction = 'sum',則可以避免除以 的操作。支援實數和複數輸入。
- 引數
size_average (bool, 可選) – 已棄用 (見
reduction)。預設情況下,損失會在批次中的每個損失元素上取平均。請注意,對於某些損失,每個樣本有多個元素。如果欄位size_average設定為False,損失則改為對每個小批次求和。當reduce為False時忽略。預設值:Truereduce (bool, 可選) – 已棄用 (見
reduction)。預設情況下,損失會根據size_average對每個小批次的觀測值取平均或求和。當reduce為False時,改為返回每個批次元素的損失,並忽略size_average。預設值:Truereduction (str, 可選) – 指定要應用於輸出的歸約方式:
'none'|'mean'|'sum'。'none':不應用歸約,'mean':輸出的總和將除以輸出中的元素數量,'sum':輸出將求和。注意:size_average和reduce正在棄用中,在此期間,指定這兩個引數中的任何一個都將覆蓋reduction。預設值:'mean'
- 形狀
輸入:,其中 表示任意數量的維度。
目標:,與輸入形狀相同。
輸出:標量。如果
reduction為'none',則為 ,與輸入形狀相同。
示例
>>> loss = nn.L1Loss() >>> input = torch.randn(3, 5, requires_grad=True) >>> target = torch.randn(3, 5) >>> output = loss(input, target) >>> output.backward()