Histogram Equalization

Is there a way in Pytorch to implement histogram equalization for tensors?

If you use torch.histc without specifying min and max, the minimum and maximum values of the data are used.
Would this be sufficient or do you need another method?

I don’t think torch.histc is differentiable, right?

Don’t think so. You might have to code up your own histc function to take the derivative.