torch.mps.set_rng_state¶ torch.mps.set_rng_state(new_state)[源代码]¶ 设置随机数生成器的状态。 Parameters new_state (torch.ByteTensor) – 所需的状态
