昇腾社区首页
中文
注册
开发者
下载

GetDynamicOutputDesc

产品支持情况

产品

是否支持

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

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

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件和库文件

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

功能说明

根据name和index的组合获取算子动态Output的TensorDesc。

函数原型

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

1
2
TensorDesc GetDynamicOutputDesc(const std::string &name, uint32_t index) const
TensorDesc GetDynamicOutputDesc(const char_t *name, uint32_t index) const

参数说明

参数名

输入/输出

描述

name

输入

算子动态Output的名称。

index

输入

算子动态Output编号,编号从1开始。

返回值说明

获取TensorDesc成功,则返回算子动态Output的TensorDesc;获取失败,则返回TensorDesc默认构造的对象,其中,主要设置DataType为DT_FLOAT(表示float类型),Format为FORMAT_NCHW(表示NCHW)。

异常处理

无。

约束说明

无。