[参数不一致]torch.cuda.get_rng_state_all

torch.cuda.get_rng_state_all

torch.cuda.get_rng_state_all()

paddle.get_rng_state

paddle.get_rng_state(device='gpu')

paddle 参数更多,并且 torch 与 paddle 的返回参数类型不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
- device 返回随机数生成器状态的设备,Pytorch 无此参数,Paddle 需设置为'gpu' 。
返回值 返回值 返回参数类型不一致, Pytorch 返回 torch.ByteTensor,Paddle 返回 GeneratorState 对象。

转写示例

返回参数类型不同

# PyTorch 写法,返回 torch.ByteTensor
x = torch.cuda.get_rng_state_all()

# Paddle 写法,返回 GeneratorState 对象
x = paddle.get_rng_state(device='gpu')