如何获取图像 torchvision.utils.save_image 保存，而不从磁盘读取它？

Question

from torchvision.utils import save_image

...
save_image(im, f'im_name.png')

就我而言（标准 mnist），使用此处的代码， im是Tensor:96 ，并且save_image有效。

我希望 memory 中的图像在其他图中显示，并且我不想在保存后将其读回，这似乎有点愚蠢。

有没有办法将生成图像和保存图像的功能分开？

---

编辑

澄清：我想要一个等价于

save_image(im, f'im_name.png')
reread = plt.imread(f'im_name.png')

无需保存图像并将其读回。 我只想要图像，我想稍后保存。 save_image function 做了一些工作，比如将多个图像堆叠成一个，将张量转换为正确大小的图像等等。 我只想要那部分而不保存到磁盘。

Answer 1

大约 2 周后，我偶然发现了这个解决方案。

grid = torchvision.utils.make_grid(im)

grid将是图像save_image刚刚保存。