oneflow.cuda.set_rng_state_all

oneflow.cuda.set_rng_state_all(new_states: Iterable[oneflow.Tensor]) → None

Sets the random number generator state of all devices.

Parameters

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