tensorplay.fork_rng

tensorplay.fork_rng(devices=None, enabled=True, _caller='fork_rng', _devices_kw='devices')[source]

Forks the RNG state: code inside the context gets a pristine RNG.

Saves the CPU RNG state and the RNG state of every CUDA device listed in devices on entry and restores them on exit, so random operations inside the block do not advance the outer streams. devices=[] forks only the CPU generator.