今天小编就为大家分享一篇pytorch实现mnist数据集的图像可视化及保存，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

如何将pytorch中mnist数据集的图像可视化及保存 导出一些库 import torch import torchvision import torch.utils.data as Data import scipy.misc import os import matplotlib.pyplot as plt BATCH_SIZE = 50 DOWNLOAD_MNIST = True 数据集的准备 #训练集测试集的准备 train_data = torchvision.dataset