SetOutputTensorShapes

Function Usage

Sets the shape of the output tensor.

Prototype

void SetOutputTensorShapes(const std::vector<MxBase::TensorDesc>& outputTensors);

Parameter Description

Parameter

Input/Output

Description

outputTensors

Input

Shape set of the output tensor.