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