Pytorch入门教程：手把手教你搭建第一个神经网络模型

本文是PyTorch入门教程，通过搭建基于MNIST数据集的全连接神经网络（MLP）模型，讲解核心操作。首先安装PyTorch（CPU/GPU版），使用torchvision加载MNIST数据集，经ToTensor转换为张量、Normalize标准化后，用DataLoader批量处理（batch_size=64）。模型定义为输入层784（28×28图像展平）、隐藏层128（ReLU激活）、输出层10（Softmax）的MLP，继承nn.Module实现前向传播。损失函数选CrossEntropyLoss，优化器用SGD（lr=0.01）。训练5个epoch，循环执行前向传播、损失计算、反向传播与参数更新，每100batch打印损失。测试时模型设为eval模式，禁用梯度计算，计算测试集准确率。教程还建议扩展方向，如调整网络结构、更换优化器或数据集等。