python - tf.histogram_fixed_width() 是否支持反向传播？

Question

我想使用 CNN 输出的直方图来计算损失。我想知道 tf.histogram_fixed_width() 是否支持渐变流回其前一层。只有它有效，我可以在计算直方图后添加一个损失层。

Answer 1

tf.histogram_fixed_width() 不支持自动梯度功能，因为直方图不是连续微分函数。您可以查看以下示例，该示例返回无梯度。

import keras.backend as K
import tensorflow as tf

value_range = [0.0, 5.0]
a = np.array([-1.0, 0.0, 1.5, 2.0, 5.0, 15])

x = K.variable(a)
hist = tf.histogram_fixed_width(x, value_range, nbins=5, dtype=tf.float32)
gradient = K.gradients(hist, x)

# output is [None]