pytorch

dd620c45 - add type annotation to distributions.kl_divergence (#78432)

add type annotation to distributions.kl_divergence (#78432) Fixes #78431 Pull Request resolved: https://github.com/pytorch/pytorch/pull/78432 Approved by: https://github.com/fritzo, https://github.com/ejguan