直接把tensor保存为图像

from  torchvision import utils as vutils
vutils.save_image(disposed_img, './test.jpg', normalize=True)

这里的disposed_img的tensor形状可以是(1, 1, w, h)，可以是(1, 3, w, h)， 也可以是(3, w, h)和(1, w, h)

参考：https://blog.csdn.net/weixin_42544131/article/details/109218882