SetOriginalShape

函数功能

设置aclTensor的OriginShape。

OriginShape表示aclTensor在经历transdata节点前（如果存在该节点）的原始shape信息，即tensor的shape数学描述。

函数原型

void SetOriginalShape(const op::Shape &shape)

参数说明

参数	输入/输出	说明
shape	输入	数据类型为op::Shape（即gert::Shape），记录了一组shape信息，例如一个三维shape：[10, 20, 30]。

返回值说明

无

约束说明

无

调用示例

      
           // 将input的OriginShape设置为[1, 2, 3, 4, 5]
void Func(aclTensor *input) {
    gert::Shape newShape;
    for (int64_t i = 1; i <= 5; i++) {
        newShape.AppendDim(i);
    }
    input->SetOriginalShape(newShape);
}

父主题： common_types