设置Tensor的原始format。
void SetOriginFormat(ge::Format origin_format)
参数 |
输入/输出 |
说明 |
|---|---|---|
origin_format |
输入 |
原始format。 关于ge::Format类型的定义,请参见ge::Format。 |
无。
无。
Tensor t = {{}, {}, {}, {}, nullptr};
t.SetOriginFormat(ge::FORMAT_NHWC);
t.SetStorageFormat(ge::FORMAT_NC1HWC0);
auto fmt = t.GetOriginFormat(); // ge::FORMAT_NHWC