设置后处理输入张量的形状为模型输出张量的形状。当前为内部使用类,请用户不要使用。
APP_ERROR SetPoolSize(const std::vector<std::vector<int>>& outputTensorShapes);
参数名 |
输入/输出 |
说明 |
---|---|---|
outputTensorShapes |
输入 |
模型输出张量的形状。 |
数据结构 |
说明 |
---|---|
APP_ERROR |
程序执行返回的错误码,请参考“MxBase/ErrorCode/ErrorCode.h”文件。 |