torch.cpu.current_device

torch.cpu.current_device()

返回 CPU 的当前设备。始终为 ‘cpu’。

注意：此函数仅用于方便设备无关的代码

返回类型

str