繁体   English   中英

PyTorch 是否具有类似 RandomState 的 object 用于随机数生成?

[英]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.

 
粤ICP备18138465号  © 2020-2023 STACKOOM.COM