0

我对 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 变量,以便以后分析它们。我该怎么做呢?谢谢你!

4

0 回答 0