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

torch.nn.Module.cuda

torch.nn.Module.cuda(device=None)

paddle.nn.Layer.to

paddle.nn.Layer.to(device="gpu")

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

参数映射

PyTorch PaddlePaddle 备注
device device PyTorch 为设备编号，Paddle 为 gpu:设备编号，需要转写。

转写示例

device 参数：设备

# PyTorch 写法:
module = torch.nn.Module()
module.cuda(0)

# Paddle 写法:
module = paddle.nn.Layer()
module.to("gpu:0")