SetOriginalShape

函数功能

设置aclTensor的OriginShape。

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

函数原型

void SetOriginalShape(const op::Shape &shape)

参数说明

返回值说明

无

约束说明

无

调用示例

// 将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);
}