dataflow.alloc_tensor
函数功能
根据shape、data type以及对齐大小申请dataflow tensor。
函数原型
1 | 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。
约束说明
无
父主题： DataFlow构图接口