Implementing a vanilla GAN in TensorFlow