\u200E

[ 输入参数用法不一致 ]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")

          