torch.mps.set\_rng\_state ========================= .. currentmodule:: torch.mps .. autofunction:: set_rng_state