[输入参数类型不一致]torch.Tensor.cuda¶

torch.Tensor.cuda ¶

torch.Tensor.cuda(device=None, non_blocking=False, memory_format=torch.preserve_format)

paddle.Tensor.cuda(device_id=None, blocking=False)

两者功能一致但参数类型不一致，具体如下：

PyTorch	PaddlePaddle	备注
device	device_id	目标 GPU 设备，输入参数类型不一致，需要转写。
non_blocking	blocking	是否同步或异步拷贝，PyTorch 和 Paddle 取值相反，需要转写。
memory_format	-	表示内存格式， Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。

# PyTorch 写法
tensor.cuda(non_blocking=True)

# Paddle 写法
tensor.cuda(blocking=False)

# PyTorch 写法
tensor.cuda("cuda:0")

# Paddle 写法
tensor.cuda(0)

# PyTorch 写法
tensor.cuda(0)

# Paddle 写法
tensor.cuda(0)