tensor类型保存图片

        os.mkdir('/home/ubuntu/DEKR-main1/11/'+str(i))
        torchvision.utils.save_image(image,'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(i)+'.png')
        for k in range(19):
            torchvision.utils.save_image(heatmap[0,k,:,:],'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(k)+'h'+'.png')
        for k in range(38):
            torchvision.utils.save_image(offset[0,k,:,:],'/home/ubuntu/DEKR-main1/11/'+str(i)+'/'+str(k)+'off'+'.png')