Does this trick apply for multilabel classification?

[riven314]

Great work!
From the softmax setting, it seems the trick is typically applied on multi-class classification.
Do you think the trick can be extended to multi-label classification? If so, what would be a possible way to extend that?