【Pytorch】torchvision.utils.save_image直接保存tensor爲圖片

一般來說，需要將tensor轉變爲numpy類型的數組從而保存圖片，這樣的過程比較繁瑣，Pytorch提供了save_image()函數，可直接將tensor保存爲圖片，若tensor在cuda上也會移到CPU中進行保存。 ) 參數： tensor (Tensor or list): Image to be saved. If given a mini-batch tensor, saves th

>>阅读原文<<