aclnnIndexAdd
产品支持情况
| 产品 | 是否支持 |
|---|---|
| [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object] | √ |
| [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object] | √ |
| [object Object]Atlas 200I/500 A2 推理产品[object Object] | × |
| [object Object]Atlas 推理系列产品 [object Object] | √ |
| [object Object]Atlas 训练系列产品[object Object] | √ |
功能说明
算子功能：在指定维度上，根据给定的索引，将源张量中的值加到输入张量中对应位置的值上。
函数原型
每个算子分为undefined，必须先调用“aclnnIndexAddGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnIndexAdd”接口执行计算。
aclnnStatus aclnnIndexAddGetWorkspaceSize(const aclTensor* self, const int64_t dim, const aclTensor* index, const aclTensor* source, const aclScalar* alpha, aclTensor* out, uint64_t* workspaceSize, aclOpExecutor** executor)aclnnStatus aclnnIndexAdd(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, aclrtStream stream)
aclnnIndexAddGetWorkspaceSize
参数说明：
self（aclTensor*，计算输入）：输入张量，Device侧的aclTensor，数据类型需要和source一致。支持undefined，undefined支持ND，数据维度支持0-8维。
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL和BFLOAT16。
dim（int64_t，计算输入）: 指定的维度。数据类型支持INT64，取值范围为[-self.dim(), self.dim()-1]。
index（aclTensor*，计算输入）：索引，Device侧的aclTensor，index的shape大小和source在dim维度上的shape值需要相等，数据类型支持INT64、INT32。undefined支持ND，数据维度支持0-1维。
source（aclTensor*，计算输入）：源张量，Device侧的aclTensor，数据类型需要和self一致，source的shape除dim维度之外，其他维度的值需要与self的shape相等。支持undefined，undefined支持ND，数据维度支持0-8维。
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL和BFLOAT16。
alpha（aclScalar*，计算输入）：host侧的aclScalar，数据类型可转换成self与source推导后的数据类型。
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL和BFLOAT16。
out（aclTensor*，计算输出）：输出Tensor，Device侧的aclTensor。shape需要与self的shape一致，数据类型需要和self一致，支持undefined，undefined支持ND，数据维度支持0-8维。
- [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：数据类型支持FLOAT、FLOAT16、INT32、INT16、INT8、UINT8、DOUBLE、INT64、BOOL和BFLOAT16。
workspaceSize（uint64_t*，出参）：返回需要在Device侧申请的workspace大小。
executor（aclOpExecutor**，出参）：返回op执行器，包含了算子计算流程。
返回值：
aclnnStatus：返回状态码，具体参见undefined。
[object Object]
aclnnIndexAdd
参数说明：
workspace（void*，入参）：在Device侧申请的workspace内存地址。
workspaceSize（uint64_t，入参）：在Device侧申请的workspace大小，由第一段接口aclnnIndexAddGetWorkspaceSize获取。
executor（aclOpExecutor*，入参）：op执行器，包含了算子计算流程。
stream（aclrtStream，入参）：指定执行任务的Stream。
返回值：
aclnnStatus：返回状态码，具体参见undefined。
约束说明
index输入取值范围在[0，self.shape[dim])范围内，即索引值输入范围为self在dim维度上的shape大小。
调用示例
示例代码如下，仅供参考，具体编译和执行过程请参考undefined。