torch.mps.get_rng_state¶ torch.mps.get_rng_state(device='mps')[source][source]¶ 返回隨機數生成器狀態作為 ByteTensor。 引數 device (torch.device 或 int, 可選) – 返回 RNG 狀態的裝置。預設值:'mps' (即 torch.device('mps'),當前的 MPS 裝置)。 返回型別 Tensor