Как ввести случайный шум в ядро ​​тренировки Op в TensorFlow?

Question

Я попытался улучшить один из оптимизаторов RMSPropOptimzer, введя случайный шум после нормального распределения. Поэтому я попытался напрямую перестроить исходный файл tensorflow\core\kernels\training_ops.cc, в частности переписывая алгоритм ApplyRMSProp.

Поэтому я попытался ввести случайный шум. Я обнаружил, что есть RandomUniform в файле ядра random_op.cc, который, похоже, генерирует нормальное распределение. Но я не знаю, как вызвать этот интерфейс в файле ядра training_ops.cc. Поэтому я спрашиваю, может ли кто-нибудь дать мне совет о том, как вводить случайную переменную в Training Op.

Answer 1

Я попытался вызвать функцию random_uniform в моем оптимизаторе на from tensorflow.python.ops import random_ops, так что тензор, содержащий случайные величины, подается на тренировку Op. И это разрешено.