Wie man zufälliges Rauschen in den Kernel von Training Op in TensorFlow einbringt?

Question

Ich habe versucht, einen der Optimierer RMSPropOptimzer zu verbessern, indem ich zufälliges Rauschen nach der normalen Verteilung injiziert. Also habe ich versucht, die Quelldatei von tensorflow\core\kernels\training_ops.cc direkt neu zu erstellen, speziell den Algorithmus von ApplyRMSProp umschreiben.

Daher versuchte ich, Zufallsrauschen einzuführen. Ich finde, dass es eine RandomUniform in der Kerneldatei random_op.cc gibt, die Normalverteilung zu generieren scheint. Aber ich weiß nicht, wie ich diese Schnittstelle in der Kernel-Datei von training_ops.cc aufrufen kann. Also frage ich, ob jemand mir einen Rat geben kann, wie man Zufallsvariable in Training Op einführt.

Answer 1

Ich habe versucht, die random_uniform Funktion in meinem Optimierer, durch from tensorflow.python.ops import random_ops, so dass ein Tensor mit zufälligen Variablen zu Training Op zugeführt werden. Und es ist gelöst.