使用 for 循环从 Tensor 保存图像

如何解决使用 for 循环从 Tensor 保存图像

我一直在运行 SRGAN 来生成超分辨率图像，但我一直在为最后一步而苦苦挣扎，我已将其命名为。

使用下面的代码，我可以从包含大量测试图像的张量中保存一张 jpeg 图像。 我想要做的是将所有测试过的图像保存为 jpeg 文件。

save_image(torch.cat([generated_image.cpu()],0),"asset/result1.jpg")

我认为解决方案是使用循环，所以我写了一个如下但它不起作用：

for i in range(len(generated_image)): 
    save_image(torch.cat([generated_image.cpu()],"asset/result1.jpg")

如果有人能就如何修复它提出建议或向我建议替代方案，我将不胜感激。

在此先非常感谢您！