如何将MNIST数据集数据图像可视化

问题

怎么将MNIST数据集里的数字图像数据可视化？

方法

基于pytorch框架，torchvision.datasets.MNIST导入数据集，对图片数据进行预处理以及使用matplotlib可视化。

步骤:

1. import 文件
   from torchvision import datasets
   import matplotlib.pyplot as plt
   import os
   import numpy as np
2. 导入数据
   #训练集
   train_ds =datasets.MNIST(
       root='data',#说明数据集下载的路径
       download=True,
       train=True
   )
   #测试集
   test_ds=datasets.MNIST(
       root='data',
       download=True,
       train=False
3. 可视化

ds_images,ds_targets=(test_ds._load_data())

fig = plt.figure()#生成图框

for i