我有两个张量,
prob_a
和prob_b
,形状为[None, 1000]
,我想计算从prob_a
到prob_b
的KL散度。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
a
和b
在最后一个轴上似乎相加为1,而不是在第一个轴上。 - Luca Di LielloAttributeError: module 'tensorflow' has no attribute 'distributions'
- jtlz2