如何计算KL散度RelaxedOneHotCategorical

Question

我想计算PyTorch中2个RelaxedOneHotCategorial分布之间的KL散度。我感觉自己做错了，因为KL差异非常大。

<CORSConfiguration xmlns="http://s3.amazonaws.com/doc/2006-03-01/">
<CORSRule>
    <AllowedOrigin>*</AllowedOrigin>
    <AllowedMethod>GET</AllowedMethod>
</CORSRule>
</CORSConfiguration>

我错过了一些琐碎的事情吗？我应该从https://arxiv.org/pdf/1611.00712.pdf做一些特别的事情吗？我看到RelaxedOneHotCategorical基于ExpConcrete并应处理下溢。