SNNL-Loss Soft Nearest Neighbor loss from paper (https://arxiv.org/pdf/1902.01889.pdf) Implemented for tf 2.x