MindSpore张量mindspore::tensor
MSTensor
#include <ms_tensor.h>
MSTensor定义了MindSpore Lite中的张量。
构造函数和析构函数
MSTensor
MindSpore Lite MSTensor的构造函数。
- 返回值
MindSpore Lite MSTensor的实例。
~MSTensor
MindSpore Lite Model的析构函数。
公有成员函数
data_type
获取MindSpore Lite MSTensor的数据类型。
TypeId在mindspore/mindspore/core/ir/dtype/type_id.h中定义。只有TypeId枚举中的数字类型或kObjectTypeString可用于MSTensor。
- 返回值
MindSpore Lite MSTensor类的MindSpore Lite TypeId。
shape
获取MindSpore Lite MSTensor的形状。
- 返回值
一个包含MindSpore Lite MSTensor形状数值的整型向量。
DimensionSize
通过参数索引获取MindSpore Lite MSTensor的维度的大小。
- 参数
- index: 定义了返回的维度的索引。
- 返回值
MindSpore Lite MSTensor的维度的大小。
ElementsNum
获取MSTensor中的元素个数。
- 返回值
MSTensor中的元素个数
Size
获取MSTensor中的数据的字节数大小。
- 返回值
MSTensor中的数据的字节数大小。
MutableData
获取MSTensor中的数据的指针。
该数据指针可用于对MSTensor中的数据进行读取和写入。
- 返回值
指向MSTensor中的数据的指针。