[参数不一致]torch.nn.Module.xpu

torch.nn.Module.xpu

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

paddle.nn.Layer.to

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

两者参数用法不一致,具体如下:

参数映射

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

转写示例

device 参数:设备

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

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