pytorch torch.nn.L1Loss_python nn.l1loss()

应用

# 1)计算绝对差总和：|0-1|+|1-1|+|2-1|+|3-1|=4；
# 2)再平均：4/4=1。
import torch
import torch.nn as nn
sample = torch.tensor([[1.,1.],[1.,1.]])
target = torch.tensor([[0.,1.],[2.,3.]])
criterion = nn.L1Loss()
loss = criterion(sample, target)

1
2
3
4
5
6
7
8
9

API

CLASS torch.nn.L1Loss(size_average=None, reduce=None, reduction: str = 'mean')
1

mean absolute error (MAE)

from torch import nn
import torch
loss = nn.L1Loss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5)
output = loss(input, target)
output.backward()
1
2
3
4
5
6
7