PyTorch 是否具有類似 RandomState 的 object 用於隨機數生成？

Question

在 numpy 我可以

import numpy as np

rs = np.random.RandomState(seed=0)

然后將 object 傳遞給周圍，例如用於依賴注入。

PyTorch有類似的接口嗎？ 我在文檔中找不到任何東西，但也許我遺漏了一些東西。

Answer 1

最接近的是torch.manual_seed ，它設置生成隨機數的種子並返回一個torch.Generator 。 This thread here有更多信息，顯然可能存在一些不一致，具體取決於您使用的是GPU還是CPU。