torch.mps.set_rng_state

torch.mps.set_rng_state(new_state, device='mps') [source]

Sets the random number generator state.

Parameters

• new_state (torch.ByteTensor) – The desired state
• device (torch.device or int, optional) – The device to set the RNG state. Default: 'mps' (i.e., torch.device('mps'), the current MPS device).