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

多分类问题在最后的输出层采用的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。