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
1 | std::vector<uint32_t> Model::GetOutputTensorShape(uint32_t index = 0); |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Index of the model output (the indexth output) |
Response Parameters
Data Structure |
Description |
|---|---|
std::vector<uint32_t> |
Shape information of the corresponding output tensor. |
Parent topic: Model