SetStorageFormat
函数功能
设置运行时Tensor的format。
函数原型
void SetStorageFormat(const ge::Format storage_format)
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
storage_format |
输入 |
运行时format。 关于ge::Format类型的定义，请参见Format。 |
返回值说明
无。
约束说明
无。
调用示例
StorageShape sh({1, 2, 3}, {1, 2, 3});
Tensor t = {sh, {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, kOnHost, ge::DT_FLOAT, nullptr};
t.SetOriginFormat(ge::FORMAT_NHWC);
t.SetStorageFormat(ge::FORMAT_NC1HWC0);
auto fmt = t.GetStorageFormat(); // ge::FORMAT_NC1HWC0
父主题： Tensor