set_cuda_rng_state

paddle. set_cuda_rng_state ( state_list ) [源代码]

设置 cuda 随机数生成器的状态信息。

参数

  • state_list (list [GeneratorState]) - 需要设置的随机数生成器状态信息列表,通过 get_cuda_rng_state() 获取。

返回

无。

代码示例

import paddle
sts = paddle.get_cuda_rng_state()
paddle.set_cuda_rng_state(sts)

使用本API的教程文档