output_shape(index)
Function Usage
Obtains the shape information of the corresponding output tensor.
Input Parameter Description
Parameter |
Type |
Description |
|---|---|---|
index |
int |
Index of the model output (the indexth output) |
Return Parameter Description
Returns the shape information (List[int]) of the output tensor.
Parent topic: Model