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.