torch.Tensor.cuda

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

返回此物件在 CUDA 記憶體中的副本。

如果此物件已在 CUDA 記憶體中並且位於正確的裝置上，則不會執行復制，並返回原始物件。

引數

• device (torch.device, optional) – 目標 GPU 裝置。預設為當前的 CUDA 裝置。

• non_blocking (bool, optional) – 如果為 True 且源在已固定記憶體中，則複製將相對於主機非同步進行。否則，該引數無效。預設值：False。

• memory_format (torch.memory_format, optional) – 返回的 Tensor 的所需記憶體格式。預設值：torch.preserve_format。