(beta)at::Device
接口原型
at::Device(const std::string &device_string)
功能描述
在安装torch_npu后,Device类型新增支持npu字段,可以从字符串描述中指示设备。
参数说明
device_string:string类型,提供的字符串必须遵循以下架构:(npu)[:<device-index>],其中npu指定设备类型,<device-index>可选,指定设备索引。
父主题: C++ API
at::Device(const std::string &device_string)
在安装torch_npu后,Device类型新增支持npu字段,可以从字符串描述中指示设备。
device_string:string类型,提供的字符串必须遵循以下架构:(npu)[:<device-index>],其中npu指定设备类型,<device-index>可选,指定设备索引。
当前产品无相关内容
未找到相关内容,请尝试其他搜索词