KL散度-相对熵

参考

在pytorch中计算KLDiv loss

注意reduction='batchmean'，不然loss不仅会在batch维度上取平均，还会在概率分布的维度上取平均。具体见官方文档

x = torch.tensor([[0.1, 0.4, 0.5]])
y = torch.tensor([[0.4, 0.2, 0.4]])

# x去拟合y的分布
kl_xy = nn.KLDivLoss(reduction='batchmean')(torch.log(x), y)
# 0.3266

# y去拟合x的分布
kl_yx = nn.KLDivLoss(reduction='batchmean')(torch.log(y), x)
# 0.2502