title: KL Divergence for two probability distributions in PyTorch
body: <p>I have two probability distributions. How should I find the KL-divergence between them in PyTorch? The regular cross entropy only accepts integer labels.</p>