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 且源位於鎖頁記憶體中，則複製相對於主機將是非同步的。否則，此引數無效。預設值：False。

• memory_format (torch.memory_format, optional) – 返回張量的期望記憶體格式。預設值：torch.preserve_format。