如何在TensorFlow的Training Op內核中引入隨機噪聲？

Question

我試圖通過在正態分佈之後注入隨機噪聲來改進優化器RMSPropOptimzer之一。所以我試圖直接重建tensorflow\core\kernels\training_ops.cc的源文件，特別是重寫了ApplyRMSProp的算法。

因此，我試圖引入隨機噪聲。我發現內核文件random_op.cc中有一個RandomUniform，這似乎會生成正態分佈。但我不知道如何在training_ops.cc的內核文件中調用該接口。所以我問，如果有人可以給我任何關於如何將隨機變量引入培訓操作的建議。

Answer 1

我試着在我的優化器中調用random_uniform函數，編號爲from tensorflow.python.ops import random_ops，這樣一個包含隨機變量的張量被饋送到訓練Op。它被解決了。