Pytorch中的CrossEntropyLoss()函数案例解读和结合one-hot编码计算Loss

1.该资源内容由用户上传，如若侵权请联系客服进行举报
2.虚拟产品一经售出概不退款（资源遇到问题，请及时私信上传者）

版权申诉

17 下载量 167 浏览量 2021-01-20 02:01:39 上传评论 3 收藏 71KB PDF 举报

温馨提示

试读

3页

使用Pytorch框架进行深度学习任务，特别是分类任务时，经常会用到如下： import torch.nn as nn criterion = nn.CrossEntropyLoss().cuda() loss = criterion(output, target) 即使用torch.nn.CrossEntropyLoss()作为损失函数。那nn.CrossEntropyLoss()内部到底是啥？？ nn.CrossEntropyLoss()是torch.nn中包装好的一个类，对应torch.nn.functional中的cross_entropy。此外，nn.CrossEntropyL

资源推荐

资源评论