跳到主要内容

SetOriginFormat

函数功能​

设置Tensor的原始format。

函数原型​

void SetOriginFormat(const ge::Format origin_format)

参数说明​

参数输入/输出说明
origin_format输入原始format。 关于ge::Format类型的定义,请参见Format。

返回值​

无

约束说明​

无

调用示例​

Tensor t = {{}, {}, {}, {}, nullptr};
t.SetOriginFormat(ge::FORMAT_NHWC);
t.SetStorageFormat(ge::FORMAT_NC1HWC0);
auto fmt = t.GetOriginFormat(); // ge::FORMAT_NHWC