GetOutputTensorShape

函数功能

获得模型输出的对应Tensor的数据shape信息。查询的结果可直接传入Tensor构造函数用来构造Tensor。

函数原型

1
std::vector<uint32_t> Model::GetOutputTensorShape(uint32_t index = 0);

参数说明

参数名

输入/输出

说明

index

输入

模型输出的索引(第index个输出)。

返回参数说明

数据结构

说明

std::vector<uint32_t>

对应输出的Tensor的shape信息。