you are right, I got confused. I need a multiclass classification loss, not a multi label. Is there an implementation of multiclass hinge loss that I can use? Is MultiMarginLoss the one that I should use?