torch.xpu.set_rng_state_all¶ torch.xpu.set_rng_state_all(new_states)[源代码]¶ 设置所有设备的随机数生成器状态。 参数 new_states (Iterable of torch.ByteTensor) – 每个设备的所需状态。
