The Generator

Learn about defining the generator class and checking the output of the untrained 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.

Get hands-on with 1200+ tech skills courses.