from torch import randint, randn
from torchmetrics.classification import MulticlassHingeLoss
metric = MulticlassHingeLoss(num_classes=3)
metric.update(randn(20, 3), randint(3, (20,)))
fig_, ax_ = metric.plot()