torch.xpu.set_rng_state¶ torch.xpu.set_rng_state(new_state, device='xpu')[原始碼][原始碼]¶ 設定指定 XPU 裝置的隨機數生成器狀態。 引數 new_state (torch.ByteTensor) – 期望的狀態 device (torch.device 或 int, 可選) – 設定 RNG 狀態的裝置。預設值: 'xpu' (即 torch.device('xpu'), 當前 XPU 裝置)。
