Error in Normalizing Network's outputs for loss calculations

Question

I have a working network called self.net, I want to normalize the outputs of that and then calculate the loss. Here what I do;

output_net1, outpu