GetOutputTensorShape
Function Usage
Obtains the shape information of the corresponding output tensor. The query result can be directly passed to the tensor constructor to construct a tensor.
Prototype
std::vector<uint32_t> GetOutputTensorShape(uint32_t index = 0)
Parameter Description
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Index of the model output (the indexth output) |
Return Parameter Description
Data Structure |
Description |
|---|---|
std::vector<uint32_t> |
Shape information of the corresponding output tensor |
Parent topic: Model