torch.mps.get_rng_state

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

以 ByteTensor 形式返回隨機數生成器狀態。

引數

device (torch.device 或 int, 可選) – 返回 RNG 狀態的裝置。預設值：'mps' (即 torch.device('mps')，當前 MPS 裝置)。

返回型別

張量