torch.cuda.set_rng_state#

torch.cuda.set_rng_state(new_state,device='cuda')[source]#

Set the random number generator state of the specified GPU.

Parameters

• new_state (torch.ByteTensor) – The desired state

• device (torch.device orint,optional) – The device to set the RNG state.Default:'cuda' (i.e.,torch.device('cuda'), the current CUDA device).