Thanks for the reply Thomas!
It makes sense that you can’t get a gradient from a histogram. However, mutual information is really an ideal metric for my network to train on. Do you have any suggestions/thoughts on a way to implement this? Or is it really not compatible with pytorch?
Thanks in advance.