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。
约束与限制
- 本接口与aclDestroyTensorList接口配套使用,分别完成aclTensorList的创建与销毁。
- 通过调用aclGetTensorListSize接口可以获取aclTensorList的大小。
调用示例
关键代码示例如下,不可以直接拷贝运行,仅供参考。
// 创建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);
父主题: 基础接口