pytorch调试过程中如何显示batch中单张图像

在进行批次训练时，有时候需要调试检查中间过程生成的图像是否正确，这里简单描述下我采用的方法。我所使用的图像为2562563大小，batch size 为100，训练时矩阵大小为（100，3，256，256），需要将其转化为（256，256，3）才能显示。如果数据在GPU中则需先转会cpu

1. 先抽取一张

aa=aa.cuda().data.cpu()
cc=aa[0]

2. 这时数据大小为（3，256，256） 类型为tensor, 这里使用permute函数，调换矩阵中各维度的顺序

cc.permute(1, 2, 0)

3.此时矩阵大小为（256，256，3）使用plt.imshow即可