tf.random.set_global_generator

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

tf.random.set_global_generator(
    generator
)

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

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

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

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