torch.cuda.get_rng_state_all

torch.cuda.get_rng_state_all()[源代码]

返回一个包含所有设备随机数状态的ByteTensor列表。

返回类型

Tensor