昇腾社区首页
中文
注册

model_gear()

函数功能

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

返回参数说明

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

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

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