Atlas 300应用(型号 3000)-推理算子输出张量描述:输出张量与输入张量形状相同的算子

时间:2023-11-01 16:13:40

输出张量与输入张量形状相同的算子

对于输出张量与输入张量形状相同的算子,可以直接将输入张量的描述插入输出张量描述所在的向量空间中。

代码示例如下所示:

v_output_desc.push_back(op.GetInputDesc(0));

其中GetInputDesc是Operator类中根据算子Input名称或者Input索引获取输入张量描述的接口,详细的接口介绍请参见《GE API参考》中的“Operator类接口”

support.huaweicloud.com/odevg-te-atlas300app3000/atlasodc_10_c30_0029.html