pytorch学习（8）：多分类问题

最新推荐文章于 2024-07-17 20:57:05 发布

jimon"铭

最新推荐文章于 2024-07-17 20:57:05 发布

阅读量853

点赞数 1

本文链接：https://blog.csdn.net/weixin_44425647/article/details/107901381

版权

本文探讨了PyTorch中多分类问题的处理，重点介绍了在输出层使用Softmax层的原因和特性。Softmax层确保每个输出值位于0到1之间，并且所有值的总和为1，这在处理概率分布时非常关键。

摘要由CSDN通过智能技术生成

多分类问题在最后的输出层采用的Softmax Layer，其具有两个特点：1.每个输出的值都是在(0,1)；2.所有值加起来和为1.

假设是最后线性层的输出，则对应的Softmax function为：

（注：使用的目的在于满足每个输出的值都为正数，即每个输出的概率值都是正的）。

import torch
from torch.utils.data import Dataset
from torch.utils.data import DataLoader
import  torch.nn.functional as F
import torch.optim as optim





batch_size = 64
transform  = transforms.Compose([
    transforms.ToTensor(),       #先将图像变换成一个张量tensor。