Loss函数的使用

import torch
from torch.nn import L1Loss
from torch import nn

inputs = torch.tensor([1,2,3],dtype = torch.float32)
target = torch.tensor([2,4,5],dtype = torch.float32)

inputs = torch.reshape(inputs,[1,1,1,3])
target = torch.reshape(target,[1,1,1,3])
loss = L1Loss()

print(loss(inputs,target))

loss1 = nn.MSELoss()
print(loss1(inputs,target))

# 交叉熵
x= torch.tensor([0.1,0.2,0.3])
y = torch.tensor([1])
x= torch.reshape(x,[1,3])
loss2 = nn.CrossEntropyLoss()
print(loss2(x,y))