下载
中文
注册

aclCreateTensorList

函数功能

创建aclTensorList,作为CANN算子API接口的入参。

aclTensorList由AscendCL框架定义,用来管理和存储由tensor类型数据构成的列表,开发者无需关注其内部实现。

函数原型

aclTensorList *aclCreateTensorList(const aclTensor *const *value, uint64_t size)

参数说明

参数名

输入/输出

说明

value

输入

Device侧的aclTensor类型的指针,其指向的值会赋给TensorList。

需提前调用aclCreateTensor接口创建aclTensor。

size

输入

张量列表的长度,取值大于0。

返回值说明

成功则返回创建好的aclTensorList,否则返回nullptr。

约束与限制

调用示例

关键代码示例如下,不可以直接拷贝运行,仅供参考。

// 创建aclTensor: input1 input2
aclTensor *input1 = aclCreateTensor(******);
aclTensor *input2 = aclCreateTensor(******);
// 创建aclTensorList
std::vector<aclTensor *> tmp{input1, input2};
aclTensorList* tensorList = aclCreateTensorList(tmp.data(), tmp.size());
// aclTensorList作为单算子API执行接口的入参
auto ret = aclnnXxxGetWorkspaceSize(srcTensor, tensorList, ..., outTensor, ..., &workspaceSize, &executor);
ret = aclnnXxx(...);
...
// 销毁aclTensorList
ret = aclDestroyTensorList(tensorList);