torch.Tensor.xpu

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

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

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

引數

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

• non_blocking (bool) – 如果為 True 且源位於固定記憶體 (pinned memory) 中，則複製相對於主機是非同步的。否則，此引數無效。預設值：False。

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