How to build a GAN using TensorFlow 2