torch.xpu.get_rng_state

torch.xpu.get_rng_state(device='xpu')

將指定 GPU 的隨機數生成器狀態作為 ByteTensor 返回。

引數

device (torch.device 或 int, 可選) – 要返回其 RNG 狀態的裝置。預設為 'xpu' (即 torch.device('xpu')，當前 XPU 裝置)。

返回型別

張量

警告

此函式將立即初始化 XPU。