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)