下载
中文
注册

GetOriginalShape

函数功能

获取aclTensor的OriginShape属性。

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

函数原型

gert::Shape GetOriginalShape()

参数说明

返回值说明

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

约束说明

调用示例

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