Vectorized KL divergence calculation between all pairs of rows of a matrix

Question

I would like to find out the KL divergence between all pairs of rows of a matrix. To explain, let\'s assume there is a matrix V of shape N x K. Now I w