device
- class torch.xpu.device(device)[source][source]
Context-manager that changes the selected device.
- Parameters
device (torch.device or int or str) – device index to select. It’s a no-op if this argument is a negative integer or
None
.
Context-manager that changes the selected device.
device (torch.device or int or str) – device index to select. It’s a no-op if
this argument is a negative integer or None
.