torch.accelerator.current_device_index

torch.accelerator.current_device_index() [исходный код]

Возвращает индекс текущего выбранного устройства для текущего ускорителя.

Возвращает:

индекс текущего выбранного устройства.

Тип возвращаемого значения:

int