blackx

摘要： 每次将tensor保存为图片时，总是要先将tensor转换到cpu，然后再转换到numpy，最后再保存图片，过程太繁琐。今天介绍个pytorch的原生api，可以直接将cuda tensor 保存为图片。 import torch import torchvision import numpy as 阅读全文