Softmax 回归的从零开始实现 pytorch

1. Softmax 回归的从零开始实现

输入为一个向量784， 也就是图片的像素长x宽。

784 = 28 * 28

keepdim表示保持维度。以0的方向相加，表示去掉row的维度，也就是结果为1row；以1的方向相加，表示去掉col的维度，也就是结果为1col。



X.reshape((-1, W.shape[0])), 维度改变为bathSize * 784， bathSize = 256

2. Softmax 回归的简洁实现 pytorch

参考

https://www.bilibili.com/video/BV1K64y1Q7wu