tensorplay.random API

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.

Returns a thread-safe random number generator for use in DataLoader workers.

This function provides a convenient way for transforms and user code to use thread-safe random number generation without manually checking worker context.

When called in a DataLoader thread worker, returns the worker’s thread-local tensorplay.Generator. When called in the main process or process workers, returns None (which causes operations to use the default global RNG).

Example

>>> from tensorplay.random import thread_safe_generator
>>> generator = thread_safe_generator()
>>> tensorplay.randint(0, 10, (5,), generator=generator)