tensor.h

概述

提供了张量相关的接口,可用于创建和修改张量信息。

起始版本:

9

相关模块:

MindSpore

汇总

类型定义

类型定义名称 描述
OH_AI_TensorHandle 指向张量对象句柄。

函数

函数名称 描述
OH_AI_TensorCreate (const char *name, OH_AI_DataType type, const int64_t *shape,
size_t shape_num, const void *data, size_t data_len)
创建一个张量对象。
OH_AI_TensorDestroy (OH_AI_TensorHandle *tensor) 释放张量对象。
OH_AI_TensorClone (OH_AI_TensorHandle tensor) 深拷贝一个张量。
OH_AI_TensorSetName (OH_AI_TensorHandle tensor, const char *name) 设置张量的名称。
OH_AI_TensorGetName (const OH_AI_TensorHandle tensor) 获取张量的名称。
OH_AI_TensorSetDataType (OH_AI_TensorHandle tensor, OH_AI_DataType type) 设置张量的数据类型。
OH_AI_TensorGetDataType (const OH_AI_TensorHandle tensor) 获取张量类型。
OH_AI_TensorSetShape (OH_AI_TensorHandle tensor,
const int64_t *shape, size_t shape_num)
设置张量的形状。
OH_AI_TensorGetShape (const OH_AI_TensorHandle tensor, size_t *shape_num) 获取张量的形状。
OH_AI_TensorSetFormat (OH_AI_TensorHandle tensor, OH_AI_Format format) 设置张量数据的排列方式。
OH_AI_TensorGetFormat (const OH_AI_TensorHandle tensor) 获取张量数据的排列方式。
OH_AI_TensorSetData (OH_AI_TensorHandle tensor, void *data) 设置张量的数据。
OH_AI_TensorGetData (const OH_AI_TensorHandle tensor) 获取张量数据的指针。
OH_AI_TensorGetMutableData (const OH_AI_TensorHandle tensor) 获取可变的张量数据指针。如果数据为空则会开辟内存。
OH_AI_TensorGetElementNum (const OH_AI_TensorHandle tensor) 获取张量元素数量。
OH_AI_TensorGetDataSize (const OH_AI_TensorHandle tensor) 获取张量中的数据的字节数大小。
OH_AI_TensorSetUserData (OH_AI_TensorHandle tensor, void *data, size_t data_size) 设置张量为用户自行管理的数据。此接口常用于复用用户数据作为模型输入,可减少一次数据拷贝。