tensorflow - tf 根据差分函数的当前梯度覆盖梯度

Question

是否可以根据当前梯度更改传播回优化器的梯度？
IE

intermediate_layer = build_intermediate(inputs)
outputs = conv2d(intermediate_layer)
#####
   ## And for all the variables inside intermediate
gradients_that_propagate_to_optimizer = true_gradients * 2

我知道 tf 有，RegisterGradient但我没有找到使用计算梯度来计算自定义梯度的方法。

标签： tensorflowbackpropagation