The Generator

The generator also needs to be changed because we’re now generating not just a larger image, but also a colour image. This means the output needs to be a 3-dimensional tensor of size (218, 178, 3).

The Generator class

Have a look at the following code for the generator’s neural network.