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

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | √ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |


#### 功能说明

设置张量数据类型。


#### 函数原型

```
set_data_type(data_type: DataType) -> Tensor
```


#### 参数说明

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| data\_type | 输入 | 要设置的数据类型。详细类型请参见DataType。 |


#### 返回值说明

(Tensor) 返回自身以支持链式调用。


#### 约束说明

- 如果data_type不是DataType类型，抛出TypeError。
- 如果设置失败，抛出RuntimeError。