torch_npu::init_npu()

torch_npu::init_npu(device_str)
torch_npu::init_npu(device_index)
torch_npu::init_npu(device)

初始化npu设备。

• 参数解释：
  • device_str：设备名称。
  • device_index：设备index，默认0。
  • device：npu设备。
• 返回值

  无

• 约束条件：

  无

• 示例：

  无