你已经了解了 PyTorch 的张量和自动求导机制,这一次,我们将迈出构建深度学习模型的第一步:用 PyTorch 实现一个全连接神经网络来识别手写数字(MNIST 数据集)。
我们将从数据准备开始,逐步构建网络结构、训练模型并评估准确率。
🧰 一、环境准备与依赖项安装
pip install torch torchvision matplotlib
📥 二、加载 MNIST 数据集(内置于 torchvision)
import torch
from torchvision import datasets, transforms
from<