《计算机视觉和图像处理简介 - 中英双语 + 代码实践版》：基于PyTorch Softmax 进行 MNIST 手写数字分类【Digit Classification with Softmax】

---

本文预估时间 25 分钟。

1. 目标与简介 Digit Classification with Softmax

Softmax 简介

多分类一种比较常用的做法是在神经网络最后一层加softmax归一化，将多个神经元的输出，映射到（0,1）区间内，可以看成概率来理解，值最大的维度所对应的位置则作为该样本对应的类。

假设我们有一个数组，V，Vi表示V中的第i个元素，那么这个元素的softmax值就是

softmax直白来说就是将原来输出是3,1,-3通过softmax函数一作用，就映射成为(0,1)的值，而这些值的累和为1（满足概率的性质），那么我们就可以将它理解成概率，在最后选取输出结点的时候，我们就可以选取概率最大（也就是值对应最大的）结点，作为我们的预测目标！