Python tf.random.Generator.split用法及代碼示例

用法

split(
    count=1
)

返回獨立 Generator 對象的列表。

兩個生成器相互獨立，因為它們生成的隨機數流沒有統計上可檢測的相關性。新生成器也獨立於舊生成器。舊生成器的狀態將被更改(與其他隨機數生成方法一樣)，因此 split 的兩次調用將返回不同的新生成器。

例如：

gens = get_global_generator().split(count=10)
for gen in gens:
  numbers = gen.normal(shape=[2, 3])
  # ...
gens2 = get_global_generator().split(count=10)
# gens2 will be different from gens

新生成器將放在當前設備上(可能與舊生成器不同)，例如：

with tf.device("/device:CPU:0"):
  gen = Generator(seed=1234)  # gen is on CPU
with tf.device("/device:GPU:0"):
  gens = gen.split(count=10)  # gens are on GPU