GetOriginalShape

函数功能

获取aclTensor的OriginShape属性。

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

函数原型

gert::Shape GetOriginalShape()

参数说明

无

返回值说明

返回值为gert::Shape，记录了一组shape信息，例如一个三维shape：[10, 20, 30]。

约束说明

无

调用示例

1
2
3

void Func(const aclTensor *input) {
    auto shape = input->GetOriginalShape();
}