Rate this Page

torch.xpu.set_rng_state_all#

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

Set the random number generator state of all devices.

Parameters

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