我对 tf.registergradient 操作有疑问
我正在使用 @tf.registergradient 对 tensorflow 渐变进行自定义更改。注意:我不是只修改 backpopogated 渐变的增量部分。这工作正常。但是现在我想保存我正在修改的变量,包括修改之前和之后,这样我可以在完成训练后分析它们。但是我不确定如何做到这一点。这是代码片段
@tf.RegisterGradient("CustomModGrad")
def noise_delta(unused_op, delta):
noiz = tf.random_normal(shape=tf.shape(grad), stddev=sigma_value)
delta = tf.add(delta,noiz)
return delta
我想存储 noiz 和 delta 变量,以便以后分析它们。我该怎么做呢?谢谢你!