SetStorageShape

函数功能

设置aclTensor的StorageShape属性。

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

StorageShape表示aclTensor在内存上的实际排布，即OriginShape实际运行时的shape格式。

函数原型

void SetStorageShape(const op::Shape &shape)

参数说明

返回值说明

无

约束说明

无

调用示例

1
2
3
4
5
6
7
8

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