关于CIFAR-10数据集向量到三维图片中的转置操作

在CIFAR-10数据集中，图片的储存方式为一维向量，如下图所示

错误方法：

a = x_data[3,:] # np.ndarray shape:(3072,)
a = a.reshape(32,32,3)
plt.imshow(a)

正确方法：

a = x_data[3,:] # np.ndarray shape:(3072,)
a = a.reshape(3,32,32).transpose(1,2,0)
plt.imshow(a)

首先使用reshape将numpy变为3x32x32的矩阵，

之后将三个维度进行转置（重新排列）为(dim1,dim2,dim0)

结果如下：