How to compute the joint-histogram in pytorch?

2

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

Thank you