SetStorageFormat

函数功能

设置aclTensor的StorageFormat。

StorageFormat表示aclTensor在内存中的排布格式，例如NCHW、ND等。

函数原型

void SetStorageFormat(op::Format format)

参数说明

返回值说明

无

约束说明

无

调用示例

1
2
3
4

// 将input的Storage Format置为ND格式
void Func(const aclTensor *input) {
    input->SetStorageFormat(ge::FORMAT_ND);
}