Ho to save all output images in CNN into a file

Question

I am running a CNN network and want to save output images (testing data) into a file. I made a function for visualization and saving. I am calling function in test_loader to