fix a bug about KL loss term#6

zxiaomzxm wants to merge 3 commits intoericjang:masterericjang/draw:masterfrom

zxiaomzxm:masterzxiaomzxm/draw:masterCopy head branch name to clipboard