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 且源位於固定記憶體中，則複製將相對於主機非同步進行。否則，該引數不起作用。預設為 False。

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