torch.cuda.set_rng_state_all

torch.cuda.set_rng_state_all(new_states)[source][source]

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

参数:

new_states (torch.ByteTensor 的可迭代对象) – 每个设备的期望状态。