Loss Functions

#L1loss:依次相减再除以个数
#Inputs(N,*)   N:batch_size多少个数据
#Targets(N,*)

# MSELOSS 均方差

#交叉熵:分类问题时 
#Inputs(N , C)    C:分类数
#Targets(N)

import torch
from torch.nn import L1Loss
from torch.nn import MSELoss
from torch.nn import CrossEntropyLoss
#from torch import nn   --->使用nn.MSELoss()


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))  #变为1batch_size,1channel,1行3列
targets = torch.reshape(targets,(1,1,1,3))

loss = L1Loss() #(reduction = 'sum')  #可以选择计算的方式reduction
result = loss(inputs,targets)  #将inputs和targets放在loss function中进行计算

loss_mse = MSELoss()
result_mse = loss_mse(inputs,targets)

print(result)#结果:tensor(0.6667)---》 [(1-1)+(2-2)+(5-3)]/3=0.6667
print(result_mse)#结果:tensor(1.3333)  ---》[(1-1)+(2-2)+(5-3)*2]/3
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值