Rate this Page

torch.mtia.current_device#

torch.mtia.current_device()[source]#

Return the index of a currently selected device.

Return type

int