交叉熵损失函数的使用目的(很肤浅的理解)

第一种使用方法

import torch
from torch import nn  # Example of target with class indices
loss = nn.CrossEntropyLoss()
input = torch.randn(3, 5, requires_grad=True)
target = torch.empty(3, dtype=torch.long).random_(5)
output = loss(input, target)
output.backward()

第二种使用方法

# Example of target with class probabilities
input = torch.randn(3, 5, requires_grad=True)
target = torch.randn(3, 5).softmax(dim=1)
output = loss(input, target)
output.backward()

 自己的理解:

 传进去的是(3,5)维度的数据,其中3可以代表有3个图片(数据),5代表有5中类别(0,1,2,3,4这几类)。

[ 0.1087, -0.4276,  0.9313, -1.0140,  2.1229]表示预测的是

       ····第一个图是第一类的概率是 0.1087

       ·····第一个图是第一类的概率是 -0.4276(负数无所谓,举的例子是随机的嘛)

。。。

target的形状就是[3],代表有三个目标真实值。其中[3,4,2]代表对应上面那个input的

----第一行的第3个值

----第二行的第4个值

----第3行的第2个值

这三个值就是真实值,表示是这些真实值的概率

交叉熵目的:

        是预测值的概率更加接近真实值,让那些真实值对于的概率的类别更加大

就是让这些红色的值变大。具体是怎么变的可以查阅相关的资料 

相关推荐

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-07-11 17:22:03       70 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-07-11 17:22:03       74 阅读
  3. 在Django里面运行非项目文件

    2024-07-11 17:22:03       62 阅读
  4. Python语言-面向对象

    2024-07-11 17:22:03       72 阅读

热门阅读

  1. 达梦数据库系列—26. DSC主备搭建

    2024-07-11 17:22:03       19 阅读
  2. Mybatis进阶の常用配置&级联查询

    2024-07-11 17:22:03       24 阅读
  3. 【MyBatis】MyBatis 理论 40 问(二)

    2024-07-11 17:22:03       22 阅读
  4. Android --- Kotlin学习之路:Okhttp 同步异步网络请求

    2024-07-11 17:22:03       19 阅读
  5. tomcat

    tomcat

    2024-07-11 17:22:03      19 阅读
  6. 探索 GraphRAG:图结构与生成式模型的融合

    2024-07-11 17:22:03       25 阅读
  7. 全国青少年信息素养大赛:排序专项:奇偶排序

    2024-07-11 17:22:03       19 阅读
  8. 主题乐园用户增长分析专项报告

    2024-07-11 17:22:03       24 阅读
  9. 使用分布式锁解决淘客返利系统中的并发问题

    2024-07-11 17:22:03       25 阅读
  10. 数据分析主流的数据分析方法与框架使用

    2024-07-11 17:22:03       21 阅读
  11. 如何修改 grafana 密码,grafana忘了怎么办

    2024-07-11 17:22:03       23 阅读