dataflow.alloc_tensor
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
函数功能
根据shape、data type以及对齐大小申请dataflow tensor。
函数原型
alloc_tensor(shape: Union[List[int], Tuple[int]], dtype, align:Optional[int] = 64) -> Tensor
参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| shape | 输入 | Tensor的shape。 |
| dtype | 输入 | Tensor的dataType。 |
| align | 输入 | 申请内存地址对齐大小，取值范围【32、64、128、256、512、1024】，默认值为64。 |
返回值
返回Tensor的实例。
异常处理
申请不到tensor指针则返回None。
约束说明
无