Gan Loss Function Pytorch