将Tensorflow中的梯度除以一个值

时间:2018-09-24 22:43:03

标签: python tensorflow

我正在尝试将Tensorflow中的梯度除以一个值(例如5),以进行偏差校正。我可以按以下方式计算和应用渐变

optimizer = tf.train.AdamOptimizer(learning_rate)
gradient,var = zip(*optimizer.compute_gradients(cost))
newgradient = #Not sure how to implement gradient/5
train_step = optimizer.apply_gradients(zip(newgradient , var))

其中成本是要最小化的误差函数。我不知道如何将梯度中的每个元素除以5(我猜梯度的形状取决于层数和每层神经元数)。

谢谢

0 个答案:

没有答案