[英]Does PyTorch have a RandomState-like object for random number generation?
在 numpy 我可以
import numpy as np
rs = np.random.RandomState(seed=0)
然后将 object 传递给周围,例如用于依赖注入。
PyTorch有类似的接口吗? 我在文档中找不到任何东西,但也许我遗漏了一些东西。
最接近的是torch.manual_seed
,它设置生成随机数的种子并返回一个torch.Generator
。 This thread here有更多信息,显然可能存在一些不一致,具体取决于您使用的是GPU还是CPU。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.