output_shape

函数功能

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

函数原型

1

output_shape(index: int)

输入参数说明

返回参数说明

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