評價此頁

torch.cuda.get_rng_state_all#

torch.cuda.get_rng_state_all()[原始碼]#

返回一個 ByteTensor 列表,表示所有裝置的隨機數狀態。

返回型別

list[torch.Tensor]