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

#### 产品支持情况

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


#### 功能说明

设置张量形状。


#### 函数原型

```
set_shape(shape: List[int]) -> TensorHolder
```


#### 参数说明

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| shape | 输入 | 形状维度列表。 |


#### 返回值说明

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


#### 约束说明

- 如果shape不是整数列表，抛出TypeError。
- 如果所有形状维度不是整数，抛出TypeError。
- 如果操作失败，抛出RuntimeError。