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_t | Tensor的大小。 |
父主题: TensorShape
operator[]
函数功能
获取Tensor指定索引维度的大小。
函数原型
size_t operator[] (uint32_t idx) const;
size_t &operator[] (uint32_t idx);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| uint32_t | 输入 | Tensor维度索引值。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | Tensor指定维度的大小。 |
父主题: TensorShape
SetValidRoi
函数功能
支持设置TensorShape的validRoi_成员变量,仅支持NHWC(N=1)、HWC与HW维度的TensorShape对象设置validRoi_成员变量, 有效区域的宽高不能超过原始图片。
函数原型
APP_ERROR SetValidRoi(Rect rect);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| rect | 输入 | 输入图像的坐标框,数据类型请参见Rect。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: TensorShape
在线提单