У меня два тензора: 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
Если нет встроенной функции, что было бы хорошим способом?