Tengo dos tensoresprob_a yprob_b con forma[None, 1000], y quiero calcular la divergencia KL deprob_a aprob_b. ¿Hay una función incorporada para esto en TensorFlow? Traté de usartf.contrib.distributions.kl(prob_a, prob_b) pero da:

NotImplementedError: No KL(dist_a || dist_b) registered for dist_a type Tensor and dist_b type Tensor

Si no hay una función incorporada, ¿cuál sería una buena solución?