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<int> shape() const
获取MindSpore Lite MSTensor的形状。
返回值
一个包含MindSpore Lite MSTensor形状数值的整型向量。
DimensionSize
virtual int DimensionSize(size_t index) const
通过参数索引获取MindSpore Lite MSTensor的维度的大小。
参数
index
: 定义了返回的维度的索引。
返回值
MindSpore Lite MSTensor的维度的大小。
ElementsNum
virtual int ElementsNum() const
获取MSTensor中的元素个数。
返回值
MSTensor中的元素个数
Size
virtual size_t Size() const
获取MSTensor中的数据的字节数大小。
返回值
MSTensor中的数据的字节数大小。
MutableData
virtual void *MutableData() const
获取MSTensor中的数据的指针。
该数据指针可用于对MSTensor中的数据进行读取和写入。
返回值
指向MSTensor中的数据的指针。