Rate this Page
★ ★ ★ ★ ★

torch.mtia.device_count#

torch.mtia.device_count()[source]#

Return the number of MTIA devices available.

Return type:

int

On this page
Show Source
PyTorch Libraries