я есть два тензора,prob_a а такжеprob_b с формой[None, 1000]и я хочу вычислить расхождение KL отprob_a вprob_b, Есть ли встроенная функция для этого в TensorFlow? Я пытался с помощьюtf.contrib.distributions.kl(prob_a, prob_b) но это дает:

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

Если нет встроенной функции, что может быть хорошим решением?