函数功能

根据算子的输出索引，获取对应输出的张量描述符，即TensorDesc。

函数原型

const CompileTimeTensorDesc *GetOutputDesc(size_t index) const

参数说明

参数	输入/输出	说明
index	输入	算子IR原型定义中的输出索引，从0开始计数。

返回值说明

输出TensorDesc的指针，当输入index非法时，返回空指针。

关于CompileTimeTensorDesc的定义，请参见CompileTimeTensorDesc类。

约束说明

无。

调用示例

// 假设已存在KernelContext *context
auto extend_context = reinterpret_cast<ExtendedKernelContext *>(context);
for (size_t idx = 0; idx < extend_context->GetComputeNodeInfo()->GetOutputsNum(); ++idx) {
  auto output_td = extend_context->GetOutputDesc(idx);
  ...
}