d5b140e5创建于 2025年12月26日历史提交

GetInputDesc

产品支持情况

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√

头文件和库文件

头文件：#include <graph/operator.h>
库文件：libgraph.so

功能说明

根据算子Input名称或Input索引获取算子Input的TensorDesc。

函数原型

说明

数据类型为string的接口后续版本会废弃，建议使用数据类型为非string的接口。

TensorDesc GetInputDesc(const std::string &name) const
TensorDesc GetInputDescByName(const char_t *name) const
TensorDesc GetInputDesc(uint32_t index) const

参数说明

参数名	输入/输出	描述
name	输入	算子Input名称。当无此算子Input名称时，则返回TensorDesc默认构造的对象，其中，主要设置DataType为DT_FLOAT（表示float类型），Format为FORMAT_NCHW（表示NCHW）。
index	输入	算子Input索引。当无此算子Input索引时，则返回TensorDesc默认构造的对象，其中，主要设置DataType为DT_FLOAT（表示float类型），Format为FORMAT_NCHW（表示NCHW）。

参数名

输入/输出

描述

name

输入

算子Input名称。

当无此算子Input名称时，则返回TensorDesc默认构造的对象，其中，主要设置DataType为DT_FLOAT（表示float类型），Format为FORMAT_NCHW（表示NCHW）。

index

输入

算子Input索引。

当无此算子Input索引时，则返回TensorDesc默认构造的对象，其中，主要设置DataType为DT_FLOAT（表示float类型），Format为FORMAT_NCHW（表示NCHW）。

返回值说明

算子Input的TensorDesc。

异常处理

无。

约束说明

无。