input_shape
函数功能
获得模型输入的对应Tensor的数据shape信息。
函数原型
1 | input_shape(index: int) |
输入参数说明
参数名 |
类型 |
说明 |
|---|---|---|
index |
int |
模型输入的索引(第index个输入)。 |
返回参数说明
返回对应输入的Tensor的shape信息,List[int]类型。
父主题: Model
获得模型输入的对应Tensor的数据shape信息。
1 | input_shape(index: int) |
参数名 |
类型 |
说明 |
|---|---|---|
index |
int |
模型输入的索引(第index个输入)。 |
返回对应输入的Tensor的shape信息,List[int]类型。