数据加载器中数据的类型维度是[B, C, H, W]，我们每次只拿一个数据出来就是图像[C, H, W]，而matplotlib.pyplot.imshow要求的显示输入维度是[H, W, C]，所以我们需要交换一下数据维度，类型把通道数放到最后面。图像

一、显示【pytorch】带batch的类型tensor类型图像显示

1、tensor转成numpy进行transpose，图像调用pyplot显示。显示类型

import matplotlib.pyplot as pltimport numpy as np#