动手深度学习笔记(十四)3.7. softmax回归的简洁实现
3. 线性神经网络
3.7. softmax回归的简洁实现
在 3.3节中, 我们发现通过深度学习框架的高级API能够使实现
线性回归变得更加容易。 同样,通过深度学习框架的高级API也能更方便地实现softmax回归模型。 本节如在 3.6节中一样, 继续使用Fashion-MNIST数据集,并保持批量大小为256。
import torch
from torch import nn
from d2l import torch as d2l
batch_size = 256
train_iter, test_i