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