Supervised Contrastive Loss, py takes features (L2 normalized) and labels as input, and return the loss.