torch.nn中常用的损失函数及使用方法

1)**torch.nn.MSELoss():**使用均方差函数对损失值进行计算
定义类时不需要传入参数,调用的时候需要传入参数

import torch
from torch.autograd import Variable
x = Variable(torch.randn(100, 100))
y = Variable(torch.randn(100, 100))
loos_f = torch.nn.MSELoss()#定义
loss = loos_f(x, y)#调用

2)torch.nn.L1Loss()类:使用平均绝对误差对损失值进行计算
定义类时不需要传入参数,在使用时要传入两个维度一样的参数:

import torch
from torch.autograd import Variable
x = Variable(torch.randn(100, 100))
y = Variable(torch.randn(100, 100))
loos_f = torch.nn.L1Loss()
loss = loos_f(x,y)

3)torch.nn.CrossEntropyLoss().使用计算交叉嫡
定义类时不需要传递参数,使用实例时需要输入两个满足交叉嫡的计算条件的参数:

import torch
from torch.autograd import Variable
x = Variable(torch.randn(100, 100))
y = Variable(torch.LongTensor(3).random_(5))#3个0-5之间的随机数字
loos_f = torch.nn.CrossEntropyLoss()
loss = loos_f(x,y)
  • 3
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值