0
我正在讀取來自外部源的梯度值(即計算在其他地方完成,但我想在「主」網絡中累積不同的源),並且我想僅使用tensorflow中的apply_gradients()
操作。問題是,漸變會以浮動形式發送。有沒有什麼辦法可以使用float數組來應用梯度和內置的Optimizer
函數?Tensorflow:如何手動編輯梯度值
在一個非常簡單的示例/測試用例中,這是我基本上想要做的。
W = tf.Variable(1.0)
b = tf.Variable(2.0)
trainable_variables = [W, b]
gradients = [0.05, 0.01] # Example gradients for W, b
# ... Somehow make this gradient vector into a tensor
optimizer.apply_gradients(zip(gradients_tensor, trainable_variables))
你的問題不清楚,增加更多信息和示例代碼。 –
好點,增加了我正在尋找的一個非常簡單的例子。 –
你有沒有試過[tf.convert_to_tensor](https://www.tensorflow.org/api_docs/python/tf/convert_to_tensor)? – aseipel