weixin_51472948的博客

原创 CNN(卷积神经网络)从0-1数字识别

定义超参数batch_size = 128 # 每个批次（batch）的样本数# 对输入的数据进行标准化处理# transforms.ToTensor() 将图像数据转换为 PyTorch 中的张量（tensor）格式，并将像素值缩放到 0-1 的范围内。# 这是因为神经网络需要的输入数据必须是张量格式，并且需要进行归一化处理，以提高模型的训练效果。