评价此页

torch.cuda.set_rng_state#

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

设置指定 GPU 的随机数生成器状态。

参数
  • new_state (torch.ByteTensor) – 所需的状态

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