torch.Generator

torch.Generator是PyTorch中用于生成随机数的类,可以指定种子生成固定的随机数。 需要注意的是,在使用过一次该随机数后,需要再次生成同样的随机数时需要再次调用该类生成,否则得到的结果不一样。 也就是说重复使用generator时,其产生的随机效果不一样(其产生的随机序列是一致的),要想重复同样的结果,需要每次调用生成的方法时,重新指定一下generator。