MindSpore接口mindspore::api

最新推荐文章于 2024-04-12 11:40:30 发布

wujianming_110117

最新推荐文章于 2024-04-12 11:40:30 发布

阅读量478

点赞数

分类专栏：智能推理视频推理 GPU，NPU，XPU，DPU，MPU

吴建明

本文链接：https://blog.csdn.net/wujianing_110117/article/details/113030387

版权

本文介绍了MindSpore的Context类，包括获取DeviceTarget和DeviceID，以及如何配置目标Device。同时，讲解了Serialization类的LoadModel方法，用于加载模型文件。Model类用于构建和执行模型，提供了Build、Predict、GetInputsInfo和GetOutputsInfo等关键函数，以及Tensor类，用于处理模型中的数据，包括数据类型、形状和内存管理。

摘要由CSDN通过智能技术生成

MindSpore接口mindspore::api
Context
#include <context.h>
Context类用于保存执行中的环境变量。
静态公有成员函数
Instance
static Context &Instance();
获取MindSpore Context实例对象。
公有成员函数
GetDeviceTarget
const std::string &GetDeviceTarget() const;
获取当前目标Device类型。
• 返回值
当前DeviceTarget的类型。
GetDeviceID
uint32_t GetDeviceID() const;
获取当前Device ID。
• 返回值
当前Device ID。
SetDeviceTarget
Context &SetDeviceTarget(const std::string &device_target);
配置目标Device。
• 参数
o device_target: 将要配置的目标Device，可选有kDeviceTypeAscend310、kDeviceTypeAscend910。
• 返回值
该MindSpore Context实例对象。
SetDeviceID
Context &SetDeviceID(uint32_t device_id);
获取当前Device ID。
• 参数
o device_id: 将要配置的Device ID。
• 返回值
该MindSpore Context实例对象。
Serialization
#include <serialization.h>
Serialization类汇总了模型文件读写的方法。
静态公有成员函数