MutableOutputTdInfo
Function Usage
Obtains the output tensor description of an operator. Note that the shape information that cannot be determined during compilation is not included in the tensor description.
Prototype
1 | CompileTimeTensorDesc *MutableOutputTdInfo(const size_t index) const |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
index |
Input |
Operator output index, starting from 0. |
Returns
Tensor description. The returned object is not const.
Constraints
None
Examples
1 2 3 4 | for (size_t i = 0; i < node->GetAllOutDataAnchorsSize(); ++i) { auto td = compute_node_info.MutableOutputTdInfo(i); ... } |
Parent topic: ComputeNodeInfo