深度学习入门笔记(7)—— Multinomial Logistic Regression / Softmax Regression
首先介绍一个非常著名的多分类数据集MNIST,也就是0到9的手写数字数据集。每个图像都是28*28,用于Pytorch数据读取的格式是NCHW,即Number、Channel、Height、Weight。读取图像之后,就能看到一个只有单通道的(灰度)图像,实际上就是一行行像素值的组合,用于SoftmaxRegression时输入得是一个向量,所以要将一行行的像素进行拼接,成为一个长的向量。同时,将