torch.mps.set_rng_state¶ torch.mps.set_rng_state(new_state, device='mps')[source][source]¶ 設定亂數產生器狀態。 參數 new_state (torch.ByteTensor) – 想要的狀態 device (torch.device 或 int, 可選) – 設定 RNG 狀態的裝置。預設值：'mps' (即 torch.device('mps')，目前的 MPS 裝置)。
