PyTorch是否具有类似RandomState的对象用于生成随机数？

Question

我可以用numpy表示

import numpy as np

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

然后将该对象传递给周围，例如进行依赖注入。

PyTorch是否具有类似的界面？我在文档中找不到任何内容，但也许我丢失了一些内容。

Answer 1

最接近的是torch.manual_seed，它设置用于生成随机数的种子并返回torch.manual_seed。该线程torch.Generator具有更多信息，显然取决于您使用的是GPU还是CPU，可能存在一些不一致之处。