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)