GetOutputDesc
Description
Obtains the output TensorDesc of the operator based on its output name or output index.
Prototype
APIs using strings will be deprecated in later versions. Use APIs supporting non-string data instead.
TensorDesc GetOutputDesc(const std::string &name) const
TensorDesc GetOutputDescByName(const char_t *name) const
TensorDesc GetOutputDesc(uint32_t index) const
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
name |
Input |
Output name of the operator. If no operator output name is available, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format). |
index |
Input |
Output index of the operator. If no operator output index is available, the default TensorDesc object is returned. Set DataType to DT_FLOAT (indicating the float type) and set Format to FORMAT_NCHW (indicating the NCHW format). |
Returns
TensorDesc of the operator output
Exception Handling
None
Restrictions
None