[ 输入参数类型不一致 ]torch.cuda.set_rng_state_all¶

torch.cuda.set_rng_state_all ¶

          torch.cuda.set_rng_state_all(new_states)

         

paddle.set_rng_state ¶

          paddle.set_rng_state(state_list, device='gpu')

         

其中 PyTorch 与 Paddle 的参数类型不一致，具体如下：

参数映射¶

PyTorch	PaddlePaddle	备注
new_states	state_list	表示每个设备需要的状态，PyTorch 类型为 torch.ByteTensor 列表，Paddle 类型为 GeneratorState 列表，需要转写。
-	device	返回随机数生成器状态的设备，Paddle 取值 gpu。

转写示例¶

参数类型不同¶

           # PyTorch 写法
x = torch.cuda.get_rng_state_all()
torch.cuda.set_rng_state_all(x)

# Paddle 写法
x = paddle.get_rng_state()
paddle.set_rng_state(x, device='gpu')

          

[ 输入参数类型不一致 ]torch.cuda.set_rng_state_all¶

torch.cuda.set_rng_state_all¶

paddle.set_rng_state¶

参数映射¶

转写示例¶

参数类型不同¶

torch.cuda.set_rng_state_all ¶

paddle.set_rng_state ¶