torch.get_rng_state

torch.get_rng_state()

返回随机数生成器的状态，以 torch.ByteTensor 的形式。

注意

返回的状态仅适用于 CPU 上的默认生成器。

参见： torch.random.fork_rng() 。

返回类型:

张量