标签:tensor 交叉 0.1 0.3 torch cross 笔记 0.6 函数
交叉损失熵函数的表达式为:
假设一个三分类问题,该问题的标签为:
Person | Dog | Cat |
0 | 1 | 2 |
将一张狗的图片输入神经网络,得到输出[0.1,0.6,0.3]
则有:
我们编程来验证一下:
import torch
from torch.nn import CrossEntropyLoss
x = torch.tensor([0.1,0.6,0.3])
y = torch.tensor([1])
x = torch.reshape(x,(1,3))
loss_cross = CrossEntropyLoss()
result_cross = loss_cross(x,y)
print(result_cross)
工作台输出为:
tensor(0.8533)
计算器输出为:
标签:tensor,交叉,0.1,0.3,torch,cross,笔记,0.6,函数 来源: https://blog.csdn.net/weixin_40692714/article/details/122733408
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。