[torch 参数更多]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 相比 Paddle 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
device	device_id	目标 GPU 设备，仅参数名不一致。
non_blocking	blocking	是否使用同步拷贝，仅参数名不一致。
memory_format	-	表示内存格式， Paddle 无此参数，一般对网络训练结果影响不大，可直接删除。