关于pytorch中用plt显示tensor的问题（tensor转numpy）

问题

图像的张量结构为（C，H，W），而plt可以显示的图片格式要求（H，W，C），C为颜色通道数，可以没有。
所以问题就是将Tensor（C，H，W）=> numpy（H，W，C）

解决办法

def transimg(img):
    img = img / 2 + 0.5 # unnormalize
    npimg = img.numpy()
    npimg1 = np.transpose(npimg,(1,2,0)) # C*H*W => H*W*C
    return npimg1