The Generator

The Generator class

Let’s now think about the generator.

The forward function

Since we’re feeding it the seed and the label tensor(one-hot target label), we need to update the forward() function to combine them before passing them through the neural network.