Does tf.histogram_fixed_width() support back propagation?
I had the similar problem. There are two ways you can try: 1. After the output layer, add an extra layer to produce the histogram; 2. Use something like tf.RegisterGradient or tf.custom_gradient to define your own gradients for the operations.
I want to use the histogram of the output from a CNN to compute the loss. I am wondering whether tf.histogram_fixed_width() supports the gradient to flow back to its former layer. Only it works, I can add a loss layer after calculating the histogram.