下载
中文
注册

SetShape

函数功能

设置Tensor的形状(Shape)。

函数原型

APP_ERROR SetShape(std::vector<uint32_t> shape);

参数说明

参数名

输入/输出

说明

shape

输入

Tensor的形状。

shape所代表的元素个数需要与Tensor的原shape所代表的元素个数相同。shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912(512 * 1024 * 1024),否则函数将抛出异常。

返回参数说明

数据结构

说明

APP_ERROR

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