C++ API
torch_npu::init_npu()
torch::npu::synchronize ()
c10::npu::current_device()
Device(const std::string &device_string)
父主题:
PyTorch2.1