torch.accelerator.current_device_index

torch.accelerator.current_device_index()

Return the index of a currently selected device for the current accelerator.

返回

当前选定设备的索引。

返回类型

int