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