Rate this Page
★ ★ ★ ★ ★

torch.mtia.current_device#

torch.mtia.current_device()[source]#

Return the index of a currently selected device.

Return type:

int

On this page
Show Source
PyTorch Libraries