---
title: SetDataType
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/ascendgraphapi/atlasgeapi_07_0487.html
sourcePath: /source/zh/canncommercial/900/API/ascendgraphapi/atlasgeapi_07_0487.html
indexId: 86ee16ea567cc4093781e6e6a17768d24dadc05aa26f41f74a6848a8635ea55172
---
# SetDataType

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| 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 &SetDataType(const ge::DataType data_type)
```


#### 参数说明

| 参数名 | 输入/输出 | 说明 |
| --- | --- | --- |
| data\_type | 输入 | 数据类型。详细数据类型请参见"DataType(https://www.hiascend.comdocument/detail/zh/canncommercial/900/maintenref/basicdataapi/atlasopapi\_07\_00514.html)"。 |


#### 返回值说明

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


#### 约束说明

无


#### 调用示例

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