Pytorch Gan Implementation Optimized