题解 | 两个正态分布之间的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))
投递字节跳动等公司10个岗位