MindSpore张量mindspore::tensor

最新推荐文章于 2024-06-26 23:30:55 发布

wujianming_110117

最新推荐文章于 2024-06-26 23:30:55 发布

阅读量638

点赞数

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

吴建明

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

版权

本文档介绍了MindSpore Lite中的MSTensor类，包括其构造函数、析构函数以及获取数据类型、形状、维度大小、元素数量和数据指针的成员函数。MSTensor是MindSpore Lite中用于处理张量的基础对象。

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

MindSpore张量mindspore::tensor
MSTensor
#include <ms_tensor.h>
MSTensor定义了MindSpore Lite中的张量。
构造函数和析构函数
MSTensor
MSTensor()
MindSpore Lite MSTensor的构造函数。
• 返回值
MindSpore Lite MSTensor的实例。
~MSTensor
virtual ~MSTensor()
MindSpore Lite Model的析构函数。
公有成员函数
data_type
virtual TypeId data_type() const
获取MindSpore Lite MSTensor的数据类型。
TypeId在mindspore/mindspore/core/ir/dtype/type_id.h中定义。只有TypeId枚举中的数字类型或kObjectTypeString可用于MSTensor。
• 返回值
MindSpore Lite MSTensor类的MindSpore Lite TypeId。
shape
virtual std::vector shape() const
获取MindSpore Lite MSTensor的形状。
• 返回值
一个包含MindSpore Lite MSTensor形状数值的整型向量。
DimensionSize
virtual int DimensionSize(size_t index) const
通过参数索引获取MindSpore Lite MSTensor的维度的大小。
• 参数
o index: 定义了返回的维度的索引。
• 返回值
MindSpore Lite MSTensor的维度的大小。
ElementsNum
virtual int ElementsNum(