Rate this Page

torch.xpu.set_rng_state#

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

Set the random number generator state of the specified GPU.

Parameters
  • new_state (torch.ByteTensor) – The desired state

  • device (torch.device or int, optional) – The device to set the RNG state. Default: 'xpu' (i.e., torch.device('xpu'), the current XPU device).