Python tf.random.set_global_generator用法及代码示例

用另一个 Generator 对象替换全局生成器。

用法

tf.random.set_global_generator(
    generator
)

此函数用提供的generator 对象替换全局生成器。随机数生成器利用tf.Variable 对象来存储其状态。用户应注意 set_global_generator 如何与 tf.function 交互的注意事项：

• tf.function 对变量创建施加了限制，因此无法在 tf.function 内自由创建新的随机生成器实例。要在 tf.function 中调用 set_global_generator ，必须已经即刻地创建了生成器实例。
• tf.function 在 trace-compilation 期间捕获变量，因此编译的 f.function 不会受到 set_global_generator 的影响，如 random_test.py/RandomTest.testResetGlobalGeneratorBadWithDefun 所示。

对于大多数用例，避免在程序初始化后调用set_global_generator，而是更愿意重置现有全局生成器的状态，例如，

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