torch.mtia.current_device

torch.mtia.current_device()[源代码]

返回当前选定设备的索引。

返回类型

int