图像通道转换——tensor从[h, w, c]转为[c, h, w]

如果我目前得到的tensor是[b,h,w,c]的，想保存成图片，但是tensor保存成图片需要[b,c,h,w]格式的

用permute或者transpose就可以正常的转过去, 让之前[b,h,w,c]是的3通道，对应到[b,c,h,w]的3通道的位置上

但是用view，他只是给你拍平之后，换了一种对之前元素的截取方式，转换后的3通道和之前的3通道是不对应的

图像通道转换——从np.ndarray的[w, h, c]转为Tensor的[c, w, h]_莫邪莫急的博客-CSDN博客