文档
注册
评分
提单
论坛
小AI

GetOutputDesc

函数功能

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

函数原型

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

TensorDesc GetOutputDesc(const std::string &name) const;

TensorDesc GetOutputDescByName(const char_t *name) const;

TensorDesc GetOutputDesc(uint32_t index) const;

参数说明

参数名

输入/输出

描述

name

输入

算子Output名称。

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

index

输入

算子Output索引。

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

返回值

参数名

类型

描述

-

TensorDesc

算子Output的TensorDesc。

异常处理

无。

约束说明

无。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词