Pytorch KDE for Entropy

I’m trying to estimate a PDF from data to calculate KL divergence or entropy. Does anyone know how to do KDE in pytorch? I’d like to be able to do it on my tensors and not transfer to numpy.

Thanks!

2 Likes