---
title: AllocTensor
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/others/dataflowdevg/dfrefc_23_0102.html
sourcePath: /source/zh/canncommercial/900/others/dataflowdevg/dfrefc_23_0102.html
indexId: 7af9ec2081e834249e2420c86905fc763368a9d6cda7f4c632bbf5b687b2226f69
---
# AllocTensor

#### 产品支持情况

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


#### 函数功能

根据shape、data type和对齐大小申请Tensor。


#### 函数原型

```
std::shared_ptr<Tensor> AllocTensor(const std::vector<int64_t> &shape, DataType data_type, uint32_t align = 512U)
```


#### 参数说明

| 参数名 | 输入/输出 | 描述 |
| --- | --- | --- |
| shape | 输入 | Tensor的shape。 |
| data\_type | 输入 | Tensor的DataType 。 |
| align | 输入 | 申请内存地址对齐大小，取值范围【32、64、128、256、512、1024】，当前为预留参数。 |


#### 返回值

申请的Tensor指针。


#### 异常处理

申请不到Tensor指针则返回NULL。


#### 约束说明

无。