Registering of kl-divergence for independent distribution (#17681)
authorNicki Skafte <skaftenicki@gmail.com>
Mon, 11 Mar 2019 15:07:22 +0000 (08:07 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 11 Mar 2019 15:10:16 +0000 (08:10 -0700)
commit8045b3eb143ed70ab1873e148d266d56e0e1481f
tree96a438399420cc4b18c5d22414d6fb1d6f7056d5
parentc02369151dfed24ec62b20efcb57fd7d3efa6f64
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
test/test_distributions.py
torch/distributions/kl.py