SetFormat

产品支持情况

产品	是否支持
Atlas A3 训练系列产品 / Atlas A3 推理系列产品	√
Atlas A2 训练系列产品 / Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

头文件/库文件

头文件：#include <ge/es_tensor_holder.h>
库文件：libeager_style_graph_builder_base.so、libeager_style_graph_builder_base_static.a

功能说明

设置TensorHolder的数据格式。

函数原型

      
           EsTensorHolder &SetFormat(const ge::Format format)

参数说明

参数名	输入/输出	说明
format	输入	张量格式。详情请参见Format。

返回值说明

参数名	类型	说明
-	EsTensorHolder &	返回当前对象的引用，支持链式调用。

约束说明

无

调用示例

      
           // 1. 创建图构建器（EsGraphBuilder）
EsGraphBuilder builder("graph_name");
// 2. 添加输入节点
EsTensorHolder data0 = builder.CreateInput(0).SeFormat(FORMAT_NCHW); // 设置输入节点输出tensor的格式类型

父主题： EsTensorHolder