63352897d1db6a7a009a1097ee9b16be13fc53b3,qucumber/utils/training_statistics.py,,KL,#,111

Before Change

        num_bases = 1
        for i in range(len(space)):
            KL += (
                cplx.norm_sqr(target_psi[:, i]) * cplx.norm_sqr(target_psi[:, i]).log()
            )
            KL -= (
                cplx.norm_sqr(target_psi[:, i])

After Change

        for i in range(len(space)):
            KL += (
                cplx.norm_sqr(target_psi[:, i])
                * (cplx.norm_sqr(target_psi[:, i]) + eps).log()
            )
            KL -= (
                cplx.norm_sqr(target_psi[:, i])

Instances