Add KL divergence loss algorithm (#11238)
* Add KL divergence loss algorithm * Apply suggestions from code review --------- Co-authored-by: Tianyi Zheng <tianyizheng02@gmail.com>
A
AtomicVar committed
c919579869ae9f57d6878336af6de6bc9a001c61
Parent: ffaa976
Committed by GitHub <noreply@github.com>
on 6/3/2024, 3:15:01 AM