跳到主要内容

GetInputDesc

函数功能

根据算子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)。

返回值

类型描述
TensorDesc算子Input的TensorDesc。

异常处理

约束说明