[ paddle 参数更多 ]torch.nn.Module.cpu

torch.nn.Module.cpu

torch.nn.Module.cpu()

paddle.nn.Layer.to

paddle.nn.Layer.to(device=None, dtype=None, blocking=None)

Paddle 相比 PyTorch 支持更多其他参数,具体如下:

参数映射

PyTorch PaddlePaddle 备注
- device 移动到的设备,PyTorch 无此参数,Paddle 设置为 "cpu"。
- dtype 数据类型,PyTorch 无此参数,Paddle 保持默认即可。
- blocking 是否阻塞,PyTorch 无此参数,Paddle 保持默认即可。