前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Pytorch - Cross Entropy Loss

Pytorch - Cross Entropy Loss

作者头像
AIHGF
修改2020-06-12 16:40:06
7.5K0
修改2020-06-12 16:40:06
举报
文章被收录于专栏:AIUAI

Pytorch - Cross Entropy Loss

Pytorch 提供的交叉熵相关的函数有:

1. CrossEntropyLoss

代码语言:javascript
复制
class torch.nn.CrossEntropyLoss(weight=None, size_average=True, ignore_index=-100, reduce=True)[source]
  • 例示:
代码语言:javascript
复制

# 1D

import torch 
import torch.nn as nn

loss = nn.CrossEntropyLoss()

# input, NxC=2x3

input = torch.randn(2, 3, requires_grad=True)

# target, N

target = torch.empty(2, dtype=torch.long).random_(3)
output = loss(input, target)
output.backward()

2. KLDivLoss

代码语言:javascript
复制
class torch.nn.KLDivLoss(size_average=True, reduce=True

3. BCELoss

代码语言:javascript
复制
class torch.nn.BCELoss(weight=None, size_average=True, reduce=True)
  • 例示:
代码语言:javascript
复制
import torch 
import torch.nn as nn

sig = nn.Sigmoid()
loss = nn.BCELoss()
input = torch.randn(3, requires_grad=True)
target = torch.empty(3).random_(2)
output = loss(sig(input), target)
output.backward()

4. BCEWithLogitsLoss

代码语言:javascript
复制
class torch.nn.BCEWithLogitsLoss(weight=None, size_average=True, reduce=True)

5. MultiLabelSoftMarginLoss

代码语言:javascript
复制
class torch.nn.MultiLabelSoftMarginLoss(weight=None, size_average=True, reduce=True)

6. MultiLabelMarginLoss

代码语言:javascript
复制
class torch.nn.MultiLabelMarginLoss(size_average=True, reduce=True)

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2018年05月04日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • Pytorch - Cross Entropy Loss
    • 1. CrossEntropyLoss
      • 2. KLDivLoss
        • 3. BCELoss
          • 4. BCEWithLogitsLoss
            • 5. MultiLabelSoftMarginLoss
              • 6. MultiLabelMarginLoss
              领券
              问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档