Registering of kl-divergence for independent distribution (#17681)
Summary:
This address issue https://github.com/pytorch/pytorch/issues/13545 and implements the proposed fix together with a single test.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17681
Differential Revision: D14360161
Pulled By: ezyang
fbshipit-source-id: 427afc88e9054b5b0dc39ebbab1087b990695ea5