跳到主要内容

GetSize

SetShape

函数功能

设置Tensor的形状。

函数原型

template<typename T>
void SetShape(std::vector<T> shape);

参数说明

参数名输入/输出说明
shape输入Tensor的形状。 shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912(512 * 1024 * 1024),否则函数将抛出异常。‬

父主题: TensorShape

GetShape

函数功能

获取Tensor的具体形状。

函数原型

std::vector<uint32_t> GetShape() const;

返回参数说明

数据结构说明
std::vector<uint32_t>Tensor的形状。

父主题: TensorShape

GetSize

函数功能

获取Tensor的大小。

函数原型

uint32_t GetSize() const;

返回参数说明

数据结构说明
uint32_tTensor的大小。

父主题: TensorShape

operator[]

函数功能

获取Tensor指定索引维度的大小。

函数原型

size_t operator[] (uint32_t idx) const;
size_t &operator[] (uint32_t idx);

参数说明

参数名输入/输出说明
uint32_t输入Tensor维度索引值。

返回参数说明

数据结构说明
size_tTensor指定维度的大小。

父主题: TensorShape

SetValidRoi

函数功能

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

函数原型

APP_ERROR SetValidRoi(Rect rect);

参数说明

参数名输入/输出说明
rect输入输入图像的坐标框,数据类型请参见Rect

返回参数说明

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

父主题: TensorShape

在线提单