跳到主要内容

input_shape(index)

input_dtype(index)​

函数功能​

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

输入参数说明​

参数名类型说明
indexint模型输入的索引（第index个输入）。

返回参数说明​

返回输入Tensor的数据类型，base.dtype类型。

父主题： Model类

output_dtype(index)​

函数功能​

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

输入参数说明​

参数名类型说明
indexint模型输出的索引（第index个输出）。

返回参数说明​

返回输出Tensor的数据类型，base.dtype类型。

父主题： Model类

input_shape(index)​

函数功能​

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

输入参数说明​

参数名类型说明
indexint模型输入的索引（第index个输入）。

返回参数说明​

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

父主题： Model类

output_shape(index)​

函数功能​

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

输入参数说明​

参数名类型说明
indexint模型输出的索引（第index个输出）。

返回参数说明​

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

父主题： Model类

model_gear()​

函数功能​

获得模型支持的动态输入档位信息，支持动态Batch、动态分辨率和分档动态维度模型。

返回参数说明​

返回模型支持的输入Tensor档位信息，List[List[int]]类型。

返回数据结构为空，表示获取分档信息失败或者模型不属于函数功能中描述的三种模型之一，具体的原因可查看报错信息确认。

例如，模型为动态分辨率模型，List[int]为某一档的高、宽值。

父主题： Model类

Model类的属性列表​

属性名说明备注
input_format获得模型输入的数据组织形式。VisionDataFormat类型，NHWC或者NCHW。
input_num获得模型的输入个数。int类型。
output_num获得模型的输出个数。int类型。

父主题： Model类

在线提单