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