评价此页

torch.xpu.set_rng_state_all#

torch.xpu.set_rng_state_all(new_states)[source]#

设置所有设备的随机数生成器状态。

参数

new_states (Iterable of torch.ByteTensor) – The desired state for each device.