Skip to main content

GetValidRoi

IsWithMargin

函数功能

查询Tensor是否存在补边。

函数原型

bool IsWithMargin() const;

返回参数说明

数据结构说明
bool判断Tensor是否存在补边的结果,布尔类型。

父主题: Tensor

SetValidRoi

函数功能

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

函数原型

APP_ERROR SetValidRoi(Rect rect);

参数说明

参数名输入/输出说明
rect输入输入图像的坐标框,数据类型请参见Rect。 起始坐标(x0, y0)只能为(0, 0)。

返回参数说明

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

父主题: Tensor

GetValidRoi

函数功能

支持查询Tensor有效区域。

函数原型

Rect GetValidRoi() const;

参数说明

返回参数说明

数据结构说明
MxBase::RectRect的数据类型,请参见Rect

父主题: Tensor

TensorMalloc

函数功能

Tensor的内存申请接口,使用该接口申请的Tensor内存无需用户管理,由内部管理释放。

函数原型

static APP_ERROR TensorMalloc(Tensor &tensor)

参数说明

参数名输入/输出说明
tensor输出待申请内存的Tensor,使用不传入内存的构造函数进行构造。

返回参数说明

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

父主题: Tensor

BatchConcat

函数功能

将多个Tensor进行组batch,按照batch维组装,默认输入的每个Tensor第一维为batch维,内存连续。

函数原型

friend APP_ERROR BatchConcat(const std::vector<Tensor> &inputs, Tensor &output)

参数说明

参数名输入/输出说明
inputs输入待组batch的Tensor列表。
output输出组装好batch的Tensor。

返回参数说明

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

父主题: Tensor

Transpose

函数功能

可通过指定的一组轴维度(axes),对输入的Tensor数据进行转置处理,如未指定具体轴维度,则默认对Tensor数据进行反序转置。

该功能支持对输出数据进行内存复用,用户可通过预先申请的内存(内存大小需与输入一致)传入输出数据。

功能仅支持在Host侧执行操作,如需处理Device侧Tensor数据,请先通过ToHost接口,将Device侧数据转移到Host侧,再进行转置。

函数原型

friend APP_ERROR Transpose(const Tensor &input, Tensor &output, std::vector<uint32_t> axes = &#123;&#125;)

参数说明

参数名输入/输出说明
input输入待转置Tensor类。 维度支持2维、3维、4维,数据类型支持float32、float16、uint8。
output输出转置后的Tensor类。
axes输入转置选项,默认值为空。 如果未指定具体axes,则默认生成反序“axes”对“input”中的数据进行反序转置。 例如:三维张量默认生成反序“axes”为{2, 1, 0} 。

返回参数说明

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

父主题: Tensor

在线提单