设置Tensor的形状(Shape)。
1 | APP_ERROR Tensor::SetShape(std::vector<uint32_t> shape); |
参数名 |
输入/输出 |
说明 |
---|---|---|
shape |
输入 |
Tensor的形状。 shape所代表的元素个数需要与Tensor的原shape所代表的元素个数相同。shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912(512 * 1024 * 1024),否则函数将抛出异常。 |
数据结构 |
说明 |
---|---|
APP_ERROR |
程序执行返回的错误码,请参考APP_ERROR说明。 |