torch.xpu.set_rng_state#
- torch.xpu.set_rng_state(new_state, device='xpu')[source]#
设置指定 GPU 的随机数生成器状态。
- 参数
new_state (torch.ByteTensor) – 所需的状态
device (torch.device or int, optional) – The device to set the RNG state. Default:
'xpu'(i.e.,torch.device('xpu'), the current XPU device).