Would it be possible to use nn.
CrossEntropyLoss
to update the generator of a GAN? for example if I have 5 classes and I want to update the generator based only on the results for a specific class I get from the discriminator.
I think its similar to what Auxilliary GANs do.