跳到主要内容

GetDesc

BatchStack

函数功能

将多batch的Tensor组合成一个Tensor,相比于BatchConcat拓展了Tensor的维度。

函数原型

static APP_ERROR BatchStack(const std::vector<TensorBase> &inputs, TensorBase &output);

参数说明

参数名输入/输出说明
inputs输入Tensor列表,列表大小不超过1024。
output输出组合后的Tensor对象。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: TensorBase

BatchVector

函数功能

合并batch组合成一个Tensor。该函数实际调用BatchConcat和BatchStack,实现组batch操作,通过keepDims参数控制具体调用函数。

函数原型

static APP_ERROR BatchVector(const std::vector<TensorBase> &inputs, TensorBase &output, const bool &keepDims = false);

参数说明

参数名输入/输出说明
inputs输入多batch模型,将TensorBase对象列表作为输出。列表大小不超过1024。
output输出组batch后输出一个TensorBase对象。
keepDims输入保持其多维特性的参数。
- “true”即保持其维度特性函数内部调用函数BatchConcat实现。
- “false”即不保持维度特性函数内部调用函数BatchStack实现。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: TensorBase

GetDesc

函数功能

用于获取详细信息。

函数原型

std::string GetDesc();

返回参数说明

数据结构说明
std::string返回详情信息。

父主题: TensorBase

CheckTensorValid

函数功能

用于查看Tensor对象是否合法。

函数原型

APP_ERROR CheckTensorValid() const;

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: TensorBase

SetValidRoi

函数功能

支持设置TensorBase的有效区域,仅支持NHWC(N=1)、HWC与HW维度的TensorBase对象设置, 有效区域的宽高不能超过原始图片。

函数原型

APP_ERROR SetValidRoi(Rect rect);

参数说明

参数名输入/输出说明
rect输入表示TensorBase对象的数据有效区域,数据类型请参见Rect

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: TensorBase

在线提单