add masking from non-scaled input as well
Created by: katyagovorkova
One more thing for the CustomMSE()
layer: to mask the scaled input as well
Created by: katyagovorkova
One more thing for the CustomMSE()
layer: to mask the scaled input as well