下载
中文
注册

input_shape(index)

函数功能

获得模型输入的对应Tensor的数据shape信息。

输入参数说明

参数名

类型

说明

index

int

模型输入的索引(第index个输入)。

返回参数说明

返回对应输入的Tensor的shape信息,List[int]类型。