TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.random.set_rng_state

tensorplay.cuda.random.set_rng_state(new_state: TensorBase, device: int | str | Any = 'cuda') → None[source]

Set the random number generator state of the specified GPU.

Parameters:

  • new_state (ByteTensor) – The desired state

  • device (tensorplay.Device or int, optional) – The device to set the RNG state. Default: 'cuda' (i.e., the current CUDA device).

tensorplay.cuda.random.seed_all

Set the seed for generating random numbers to a random number on all GPUs.

tensorplay.cuda.random.set_rng_state_all

Set the random number generator state of all devices.

Ask DeepWiki