题解 | 两个正态分布之间的KL散度
两个正态分布之间的KL散度
https://www.nowcoder.com/practice/7afa7e4a9e2f428ab8ebefd6771116e5
import numpy as np def kl_divergence_normal(mu_p, sigma_p, mu_q, sigma_q): kl_div=np.log(sigma_q/sigma_p)+((sigma_p)**2+(mu_p-mu_q)**2)/(2*(sigma_q)**2)-1/2 return kl_div if __name__ == "__main__": mu_p, sigma_p, mu_q, sigma_q = map(float, input().split()) print(kl_divergence_normal(mu_p, sigma_p, mu_q, sigma_q))