torch.xpu.set_device

torch.xpu.set_device(device)

設定當前裝置。

引數

device (torch.device 或 int 或 str) – 選定的裝置。如果此引數為負數，則此函式無效。