Comment by ilzmastr

If you ever wondered about the generalization to multiple classes, there is a reason that the gans look totally different:

https://proceedings.mlr.press/v137/kavalerov20a/kavalerov20a...

It turns out 2 classes is special. Better to add the classes as side information rather than try to make it part of the main objective.