torch.mtia.get_rng_state

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

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

引數

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

返回型別

張量