我的第一次尝试

import torch import torchvision # 由流行的数据集、模型架构和用于计算机视觉的常见图像转换组成。 import torchvision.transforms as transforms # 用来进行数据与处理的模块 import torch.nn as nn # 深度学习中的损失函数封装在torch.nn中 import torch.nn.functional as F import torch.optim as optim # 深度学习中的优化方法都封装在torch.optim里面 import matplotlib.pyplot as plt import