动手深度学习笔记(十三)3.6. softmax回归的从零开始实现
3. 线性神经网络
3.6. softmax回归的从零开始实现
就像我们从零开始实现线性回归一样, 我们认为softmax回归也是重要的基础,因此你应该知道实现softmax回归的细节。 本节我们将使用刚刚在 3.5节中引入的Fashion-MNIST数据集, 并设置数据迭代器的批量大小为256。
import torch
from IPython import display
from d2l import torch as d2l
batch_size = 256