GetTensorType
TensorBaseMalloc
函数功能
用于获取TensorBase对象的内存。
Tensor对象在析构时会自动释放内存，因此无需通过函数释放内存。
函数原型
static APP_ERROR TensorBaseMalloc(TensorBase &tensor);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| tensor | 输入 | 用来获取对应内存的Tensor数据。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase
TensorBaseCopy
函数功能
内存复制函数，根据MemoryData中指定的内存位置在Host侧和Device侧之间进行复制。
函数原型
static APP_ERROR TensorBaseCopy(TensorBase &dst, const TensorBase &src);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| dst | 输出 | 复制后的目标内存。 |
| src | 输入 | 待复制的源内存。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase
GetTensorType
函数功能
用于获取Tensor的内存类型。
函数原型
MemoryData::MemoryType GetTensorType() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MemoryType | 申请的内存类型： ● MEMORY_HOST：对应Host侧。 ● MEMORY_DEVICE：对应Device侧。 ● MEMORY_DVPP：对应DVPP侧。 ● MEMORY_HOST_MALLOC：对应malloc申请内存。 ● MEMORY_HOST_NEW：对应new申请内存。 |
父主题： TensorBase
GetSize
函数功能
用于获取张量数据对应内存大小。
size大小应与实际内存大小一致，否则可能会导致程序出现coredump情况。
函数原型
size_t GetSize() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | 返回具体数值。 |
父主题： TensorBase
GetByteSize
函数功能
用于获取buffer字节数据量。
函数原型
size_t GetByteSize() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | 返回具体数值。 |
父主题： TensorBase
SetShape
函数功能
设置TensorBase的形状。
函数原型
APP_ERROR SetShape(std::vector<uint32_t> shape);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| shape | 输入 | shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912（512 * 1024 * 1024），否则函数将抛出异常。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase
在线提单