L1Loss函数的使用————PyTorch

哔哩大学的PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】
的P23讲讲述了L1Loss函数的使用。

import torch
from torch.nn import L1Loss

inputs = torch.tensor([1, 2, 3], dtype=torch.float32) # 变成浮点数
targets = torch.tensor([1, 2, 5], dtype=torch.float32)

inputs = torch.reshape(inputs, (1, 1, 1, 3))
targets = torch.reshape(targets,(1, 1, 1, 3))

loss = L1Loss()
result = loss(inputs, targets)
print(result)
# 代码的计算过程为  :
# 1-1 = 0, 2-2 = 0,3-5 = -2 , 绝对值相加除以3 = 0.6667
# 若  loss = L1Loss(reduction="sum")   则结果只相加 = 2

结果:
在这里插入图片描述

你可能感兴趣的:(pytorch,pytorch,深度学习,神经网络)