tensor.h
概述
提供了张量相关的接口,可用于创建和修改张量信息。
起始版本:
9
相关模块:
汇总
类型定义
类型定义名称 | 描述 |
---|---|
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) | 设置张量为用户自行管理的数据。此接口常用于复用用户数据作为模型输入,可减少一次数据拷贝。 |