PyTorch 是否有一个类似 RandomState 的对象来生成随机数？

Question

在 numpy 中我可以

import numpy as np

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

然后传递该对象，例如依赖注入。

PyTorch 有类似的界面吗？我在文档中找不到任何内容，但也许我遗漏了什么。

Answer 1

最接近的是 torch.manual_seed, which sets the seed for generating random numbers and returns a torch.Generator. This thread here 有更多信息，显然可能存在一些不一致，具体取决于您使用的是 GPU 还是 CPU。