https://mingredbestcheapprice.blogspot.com/2023/10/implementing-multiclass-dice-loss.html