評價此頁

torch.xpu.set_rng_state_all#

torch.xpu.set_rng_state_all(new_states)[原始碼]#

設定所有裝置的隨機數生成器狀態。

引數

new_states (Iterable of torch.ByteTensor) – 每個裝置的期望狀態。