《动手学深度学习》softmax回归的简洁实现（李沐）

3.7. softmax回归的简洁实现

import torch
from torch import nn
from d2l import torch as d2l

batch_size = 256
train_iter, test_iter = d2l.load_data_fashion_mnist(batch_size)

3.7.1. 初始化模型参数

#PyTorch不会隐式地调整输入的形状。因此，
#我们在线性层前定义了展平层（flatten），来调整网络输入的形状
net = nn.Sequential