How to compute the joint-histogram in pytorch?

1

how to compute the joint-histogram between two tensor in pytorch?
i know there is a func: torch.histc() to compute the histogram, but i also want to use the joint-histogram to calculate the mutual information for loss-backward, can anyone implement this code?

2

Hi jiawei_chen,
Did u find any solution? Any help will be appreciated.

Thank you

3

any solution?

I have been looking for a np.histogram2D equivalent in pytorch

4

any solution?

I have been looking for a np.histogram2D equivalent in pytorch