Solved by ptrblck in post #2 
              
              
                You can directly apply both mentioned losses, as they would expect the model output and target to have the same shape, which is the case for your use case. 
Unfortunately, I not sure how SSIM can be used for your use case, but if I’m not mistaken the original implementation uses 2D convs internally,…