我需要一段计算js散度的python代码