pytorch, KL散度，reduction='batchmean'

本文轉載自查看原文 2021-01-01 23:09 509 Pytorch/ PyTorch/ KL散度/ Deep Learning

在pytorch中計算KLDiv loss時，注意reduction='batchmean'，不然loss不僅會在batch維度上取平均，還會在概率分布的維度上取平均。

本站轉載的文章為個人學習借鑒使用，本站對版權不負任何法律責任。如果侵犯了您的隱私權益，請聯系本站郵箱yoyou2525@163.com刪除。

猜您在找 KL散度理解以及使用pytorch計算KL散度 ELBO 與 KL散度 KL散度淺談KL散度 KL散度的理解 KL散度與JS散度 python 3計算KL散度（KL Divergence） KL散度(KL divergence, JS divergence) 相對熵（KL散度） KL 散度（相對熵）