[ 输入参数类型不一致 ]torch.set_default_device

torch.set_default_device

torch.set_default_device(device)

paddle.set_device

paddle.set_device(device)

功能一致,参数类型不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
device device PyTorch 支持 torch.device 。PaddlePaddle 支持 str。