torch.Tensor.cuda

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

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

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

參數

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

• non_blocking (bool) – 如果為 True 且來源位於釘選記憶體 (pinned memory) 中，則複製操作將會相對於主機 (host) 非同步執行。 否則，此參數無效。預設值：False。

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