2020年5月15日
KL散度-相对熵
摘要： "参考" 在pytorch中计算KLDiv loss 注意reduction='batchmean'，不然loss不仅会在batch维度上取平均，还会在概率分布的维度上取平均。具体见 "官方文档" 阅读全文
posted @ 2020-05-15 10:37 happy_fan 阅读(2326) 评论(0) 推荐(0) 编辑