如何在TensorFlow的Training Op内核中引入随机噪声？

Question

我试图通过在正态分布之后注入随机噪声来改进其中一个优化器 RMSPropOptimzer。所以我尝试直接重建tensorflow\core\kernels\training_ops.cc的源文件，具体重写ApplyRMSProp的算法。

因此，我尝试引入随机噪声。我发现内核文件random_op.cc里面有个RandomUniform，好像是生成正态分布的。但是我不知道如何在training_ops.cc的内核文件中调用那个接口。所以我问是否有人可以给我关于如何将随机变量引入 Training Op 的建议。

Answer 1

我试图通过 from tensorflow.python.ops import random_ops 在我的优化器中调用 random_uniform 函数，以便将包含随机变量的张量提供给训练 Op。并且解决了。