我需要在tensorflow中最小化KL损失。
我尝试了这个函数
我尝试手动实现它:
我尝试了这个函数
tf.contrib.distributions.kl(dist_a, dist_b, allow_nan=False, name=None)
,但失败了。我尝试手动实现它:
def kl_divergence(p,q):
return p* tf.log(p/q)+(1-p)*tf.log((1-p)/(1-q))
这是正确的吗?