Kullback Leibler Divergence是否已在TensorFlow中实施？

Question

我正在使用tensorflow并使用神经网络来解决多标签分类问题。我使用Softmax交叉熵作为我的损失函数：

#Softmax loss
cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(pred, y)) 

现在，我认为我应该使用KL分歧损失功能。但是，我没有在tensorflow中发现它可以帮助我使用KL发散损失函数而不是Softmax损失吗？

Answer 1

干得好：

tf.contrib.distributions.kl(distribution_1, distribution_2)