pytorch(b站小土堆学习笔记P16)

损失函数和反向传播

常用loss函数

1.L1LOSS

pytorch(b站小土堆学习笔记P16)_第1张图片

loss = nn.L1Loss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5)
output = loss(input, target)
output.backward()

你可能感兴趣的:(pytorch,学习,笔记)