tf.random.set_global_generator

Заменяет глобальный генератор на другой Generator объект.

tf.random.set_global_generator(
    generator
)

Эта функция заменяет глобальный генератор предоставленным generator объектом. Генератор случайных чисел использует объект tf.Variable для хранения своего состояния. Пользователь должен учитывать особенности взаимодействия set_global_generator с tf.function:

• tf.function накладывает ограничения на создание переменных, поэтому нельзя свободно создавать новый экземпляр генератора случайных чисел внутри tf.function. Чтобы вызвать set_global_generator внутри tf.function, экземпляр генератора должен быть уже создан в режиме eager.
• tf.function захватывает переменную во время компиляции трассировки, поэтому скомпилированная f.function не будет затронута set_global_generator, как показано в random_test.py/RandomTest.testResetGlobalGeneratorBadWithDefun.

В большинстве случаев избегайте вызова set_global_generator после инициализации программы и предпочтительнее перезадать состояние существующего глобального генератора, например,

rng = tf.random.get_global_generator()
rng.reset_from_seed(30)