昇腾社区首页
中文
注册

函数:get_input_data_type

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

根据模型描述信息获取模型中指定输入的数据类型。

函数原型

  • C函数原型
    1
    aclDataType aclmdlGetInputDataType(const aclmdlDesc *modelDesc, size_t index)
    
  • python函数
    1
    data_type = acl.mdl.get_input_data_type(model_desc, index)
    

参数说明

参数名

说明

model_desc

int,aclmdlDesc类型数据的指针地址。

index

int,指定获取第几个输入的大小,“index”值从0开始。

返回值说明

返回值

说明

data_type

int,返回指定输入的数据类型,具体请参见aclDataType