torch.cuda.get_rng_state

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

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

引數

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

返回型別

張量

警告

此函式會立即初始化 CUDA。