Rate this Page
★★★★★

torch.xpu.set_device#

torch.xpu.set_device(device)[source]#

Set the current device.

Parameters:

device (torch.device orint orstr) – selected device. This function is ano-op if this argument is negative.