如何获取图像 torchvision.utils.save_image 保存，而不从磁盘读回？
How to get the image torchvision.utils.save_image saves, without reading it back from disk?
from torchvision.utils import save_image
...
save_image(im, f'im_name.png')
在我的例子中（标准 mnist），使用来自 here 的代码，im 是 Tensor:96，并且 save_image 有效。
我想让内存中的那个图像在其他图中显示，我不想在保存之后再读回来，这似乎有点愚蠢。
有没有办法将生成图像和保存图像的功能分开？
编辑
说明：
我想要等同于
save_image(im, f'im_name.png')
reread = plt.imread(f'im_name.png')
不保存图像并读回。
我只想要图像，我想稍后保存它。
save_image 函数做了一些工作，比如将多张图像堆叠成一张，将张量转换为正确大小的图像等等。我只想要那部分而不保存到磁盘。
大约 2 周后，我偶然发现了解决方案。
grid = torchvision.utils.make_grid(im)
grid 将是 save_image 刚刚保存的图像。
from torchvision.utils import save_image
...
save_image(im, f'im_name.png')
在我的例子中（标准 mnist），使用来自 here 的代码，im 是 Tensor:96，并且 save_image 有效。
我想让内存中的那个图像在其他图中显示，我不想在保存之后再读回来，这似乎有点愚蠢。
有没有办法将生成图像和保存图像的功能分开？
编辑
说明： 我想要等同于
save_image(im, f'im_name.png')
reread = plt.imread(f'im_name.png')
不保存图像并读回。
我只想要图像，我想稍后保存它。
save_image 函数做了一些工作，比如将多张图像堆叠成一张，将张量转换为正确大小的图像等等。我只想要那部分而不保存到磁盘。
大约 2 周后，我偶然发现了解决方案。
grid = torchvision.utils.make_grid(im)
grid 将是 save_image 刚刚保存的图像。